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: