fix attribute hostkeys

Author: phueb

ludwig/__main__.py

@@ -21,7 +21,9 @@
 
 def add_ssh_config():
     """
-    append contents of /media/ludwig_data/.ludwig/config to ~/.ssh/ludwig_config
+    copy content of <ludwig_data_path>/.ludwig/config to ~/.ssh/ludwig_config.
+
+    note: this file contains the hostnames of Ludwig workers and their corresponding IP addresses.
     """
     src = configs.WorkerDirs.ludwig_data / '.ludwig' / 'config'
     dst = Path().home() / '.ssh' / 'ludwig_config'  # do not overwrite existing config
@@ -135,7 +137,7 @@ def submit():
                         required=False,
                         help='Whether to upload jobs to Ludwig. Set false for testing')
     parser.add_argument('-s', '--skip-hostkey', action='store_true', dest='skip_hostkey',
-                        required=False,
+                        required=False, default=False,
                         help='Whether to skip hostkey checking. Unsafe, but may prevent SSH connection error.')
     namespace = parser.parse_args()
 

ludwig/uploader.py

@@ -87,7 +87,7 @@ def to_disk(self,
 
     def start_jobs(self,
                    worker: str,
-                   skip_hostkey: bool,  # True if skipping hostkey check (not safe)
+                   skip_hostkey: bool = False,  # True if skipping hostkey check (not safe)
                    ) -> None:
         """
         source code is uploaded.
@@ -118,12 +118,12 @@ def start_jobs(self,
         cnopts = pysftp.CnOpts()
         if skip_hostkey:
             print('WARNING: Skipping hostkey check. Known hosts will not be consulted')
-            cnopts.hostkey = None
+            cnopts.hostkeys = None
 
         # connect via sftp
         ludwig_data_path = self.project_path.parent
         private_key_path = ludwig_data_path / '.ludwig' / 'id_rsa'
-        print(f'Looking for private key path in {private_key_path}')
+        print(f'Looking for private key in {private_key_path}')
         if not private_key_path.exists():
             raise OSError(f'Did not find {private_key_path}')
         sftp = pysftp.Connection(username='ludwig',