Skip to content

Commit

Permalink
Rename ssh runner file to command runner
Browse files Browse the repository at this point in the history
  • Loading branch information
carolineechen committed Aug 29, 2024
1 parent e28812d commit 0fef4b0
Show file tree
Hide file tree
Showing 5 changed files with 7 additions and 7 deletions.
6 changes: 3 additions & 3 deletions runhouse/resources/hardware/cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -431,7 +431,7 @@ def is_shared(self) -> bool:
return f"{self._creds.name}/" in ssh_creds.get("ssh_private_key", "")

def _command_runner(self, node: Optional[str] = None) -> "CommandRunner":
from runhouse.resources.hardware.sky_ssh_runner import (
from runhouse.resources.hardware.sky_command_runner import (
SkyKubernetesRunner,
SkySSHRunner,
)
Expand Down Expand Up @@ -1214,7 +1214,7 @@ def rsync(
)
return

from runhouse.resources.hardware.sky_ssh_runner import SshMode
from runhouse.resources.hardware.sky_command_runner import SshMode

# If no address provided explicitly use the head node address
node = node or self.address
Expand Down Expand Up @@ -1449,7 +1449,7 @@ def _run_commands_with_runner(
require_outputs: bool = True,
_ssh_mode: str = "interactive", # Note, this only applies for non-password SSH
):
from runhouse.resources.hardware.sky_ssh_runner import SshMode
from runhouse.resources.hardware.sky_command_runner import SshMode

if isinstance(commands, str):
commands = [commands]
Expand Down
4 changes: 2 additions & 2 deletions runhouse/resources/hardware/on_demand_cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ def docker_user(self) -> str:
if self.launched_properties["cloud"] == "kubernetes":
return "root"

from runhouse.resources.hardware.sky_ssh_runner import get_docker_user
from runhouse.resources.hardware.sky_command_runner import get_docker_user

if not self._creds:
return
Expand Down Expand Up @@ -664,7 +664,7 @@ def ssh(self, node: str = None):

else:
# If SSHing onto a specific node, which requires the default sky public key for verification
from runhouse.resources.hardware.sky_ssh_runner import SshMode
from runhouse.resources.hardware.sky_command_runner import SshMode

sky_key = Path(
self.creds_values.get("ssh_private_key", self.DEFAULT_KEYFILE)
Expand Down
2 changes: 1 addition & 1 deletion runhouse/resources/hardware/sagemaker/sagemaker_cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -706,7 +706,7 @@ def _run_commands_with_runner(
return_codes.append((255, "", str(e)))
else:
# Host can be replaced with name (as reflected in the ~/.ssh/config file)
from runhouse.resources.hardware.sky_ssh_runner import SkySSHRunner
from runhouse.resources.hardware.sky_command_runner import SkySSHRunner

runner = SkySSHRunner(
(self.name, self.ssh_port),
Expand Down
2 changes: 1 addition & 1 deletion runhouse/resources/hardware/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -150,7 +150,7 @@ def _run_ssh_command(
ssh_private_key: str,
docker_user: str,
):
from runhouse.resources.hardware.sky_ssh_runner import SkySSHRunner
from runhouse.resources.hardware.sky_command_runner import SkySSHRunner

runner = SkySSHRunner(
ip=(address, ssh_port),
Expand Down

0 comments on commit 0fef4b0

Please sign in to comment.