take lecun offline

ludwig/__init__.py

@@ -1,6 +1,6 @@
 
 
-__version__ = '4.0.3'
+__version__ = '4.0.4'
 
 
 def print_ludwig(s):

ludwig/configs.py

@@ -12,8 +12,9 @@ class WorkerDirs:
 class Remote:
     watched_pattern = 'run*.py'  # this is required for watcher to know which file to run
     path_to_ssh_config = Path.home() / '.ssh' / 'ludwig_config'
-    online_worker_names = ['norman', 'hinton', 'pitts', 'hawkins', 'lecun', 'hoff', 'bengio']
+    online_worker_names = ['norman', 'hinton', 'pitts', 'hawkins', 'hoff', 'bengio']
     # TODO may 19: os error when connecting to hebb
+    # TODO may 27: timeout when connecting to lecun
     all_worker_names = ['hoff', 'norman', 'hebb', 'hinton', 'pitts', 'hawkins', 'bengio', 'lecun']
     group2workers = {'half1': ['hoff', 'norman', 'hebb', 'hinton'],
                      'half2': ['pitts', 'hawkins', 'bengio', 'lecun']}

ludwig/uploader.py

@@ -92,9 +92,9 @@ def start_jobs(self,
         # 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 in {private_key_path}')
         if not private_key_path.exists():
-            raise OSError(f'Did not find {private_key_path}')
+            raise OSError(f'Did not find private key in {private_key_path}')
+        print_ludwig(f'Connecting to {worker}')
         sftp = pysftp.Connection(username='ludwig',
                                  host=configs.Constants.worker2ip[worker],
                                  private_key=str(private_key_path),
@@ -141,7 +141,6 @@ def kill_jobs(self,
         # connect via sftp
         ludwig_data_path = self.project_path.parent
         private_key_path = ludwig_data_path / '.ludwig' / 'id_rsa'
-        print_ludwig(f'Looking for private key in {str(private_key_path)}')
         if not private_key_path.exists():
             raise OSError(f'Did not find private key in {private_key_path}')
         sftp = pysftp.Connection(username='ludwig',