diff --git a/ludwig/__init__.py b/ludwig/__init__.py index ba218bd..ff2b397 100644 --- a/ludwig/__init__.py +++ b/ludwig/__init__.py @@ -1,6 +1,6 @@ -__version__ = '4.0.3' +__version__ = '4.0.4' def print_ludwig(s): diff --git a/ludwig/configs.py b/ludwig/configs.py index 11fe267..55784f0 100644 --- a/ludwig/configs.py +++ b/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']} diff --git a/ludwig/uploader.py b/ludwig/uploader.py index 58b2163..4e9e277 100644 --- a/ludwig/uploader.py +++ b/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',