From b23d98982440a3d75c3eef48cb8dc2a1a16e185a Mon Sep 17 00:00:00 2001 From: Stella Biderman Date: Thu, 12 Mar 2026 12:56:03 -0400 Subject: [PATCH] Add __main__.py and fix remote pip install command --- obliteratus/__main__.py | 5 +++++ obliteratus/remote.py | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) create mode 100644 obliteratus/__main__.py diff --git a/obliteratus/__main__.py b/obliteratus/__main__.py new file mode 100644 index 0000000..995f990 --- /dev/null +++ b/obliteratus/__main__.py @@ -0,0 +1,5 @@ +"""Allow running obliteratus as ``python -m obliteratus``.""" + +from obliteratus.cli import main + +main() diff --git a/obliteratus/remote.py b/obliteratus/remote.py index 57c7be7..960c7ad 100644 --- a/obliteratus/remote.py +++ b/obliteratus/remote.py @@ -193,7 +193,7 @@ class RemoteRunner: self.on_log("Installing obliteratus on remote...") install_cmd = ( f"{self.config.python} -m pip install --quiet " - f"'obliteratus @ git+https://github.com/EleutherAI/OBLITERATUS.git'" + f"git+https://github.com/EleutherAI/OBLITERATUS.git" ) rc = self.run_ssh(install_cmd, stream=True, timeout=self.config.install_timeout) if rc != 0: