Add __main__.py and fix remote pip install command

This commit is contained in:
Stella Biderman
2026-03-12 12:56:03 -04:00
parent cbdb772eb9
commit b23d989824
2 changed files with 6 additions and 1 deletions
+5
View File
@@ -0,0 +1,5 @@
"""Allow running obliteratus as ``python -m obliteratus``."""
from obliteratus.cli import main
main()
+1 -1
View File
@@ -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: