Requirements Manager support for git packages

VoltaML · Oct 15, 2023 · 9523828 · 9523828
1 parent 0b0e683
commit 9523828
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/core/install_requirements.py b/core/install_requirements.py
@@ -17,6 +17,7 @@
     "open_clip_torch": "open_clip",
     "python-multipart": "multipart",
     "discord.py": "discord",
+    "HyperTile": "hyper-tile",
 }
 logger = logging.getLogger(__name__)
 
@@ -45,8 +46,11 @@ def install_requirements(path_to_requirements: str = "requirements.txt"):
                     continue
 
             if "git+http" in i:
-                logger.debug(f"Skipping git requirement (cannot check version): {i}")
-                continue
+                tmp = i.split("@")[0].split("/")[-1].replace(".git", "").strip()
+                logger.debug(
+                    f"Rewrote git requirement (cannot check hash, but proceeding): {i} => {tmp}"
+                )
+                i = tmp
 
             if "==" in i:
                 requirements[i.split("==")[0]] = i.replace(i.split("==")[0], "").strip()