unify SSH connectivity tests
@@ -123,14 +123,14 @@ def create():
@main.command()
def test():
from tensorhive.config import SSH
- from tensorhive.core.managers.SSHConnectionManager import SSHConnectionManager
+ from tensorhive.core.managers.TensorHiveManager import TensorHiveManager
logging.basicConfig(level=logging.INFO, format='%(message)-79s')
- # logging.getLogger('pssh').setLevel(logging.CRITICAL)
+
if not SSH.AVAILABLE_NODES:
click.echo('[!] Empty ssh configuration. Please check {}'.format(SSH.HOSTS_CONFIG_FILE))
else:
- SSHConnectionManager.test_all_connections(config=SSH.AVAILABLE_NODES, key_path=SSH.KEY_FILE)
+ TensorHiveManager.test_ssh()
@main.command()
