use gitsubmodle for dev

PrimeIntellect-ai · Jan 6, 2025 · cd353f3 · cd353f3
1 parent d42a1c0
commit cd353f3
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 3 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -1,3 +1,6 @@
 [submodule "third_party/gloo"]
 	path = third_party/gloo
 	url = https://github.com/facebookincubator/gloo.git
+[submodule "third_party/optimizers"]
+	path = third_party/optimizers
+	url = [email protected]:facebookresearch/optimizers.git
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,7 +19,7 @@ dependencies = [
     "pyarrow",
     "toposolve",
     "psutil",
-    "torch-shampoo @ git+https://github.com/facebookresearch/optimizers.git@main",
+    "torch-shampoo",
 ]
 
 [project.optional-dependencies]
@@ -46,3 +46,6 @@ line-length = 120
 
 [tool.uv]
 dev-dependencies = ["ruff>=0.5.0", "pre-commit>=3.0.0","pytest>=7.0.0", "faker"]
+
+[tool.uv.sources]
+torch-shampoo = { path = "third_party/optimizers", editable = true }
diff --git a/third_party/optimizers b/third_party/optimizers
diff --git a/uv.lock b/uv.lock