Merge pull request #136 from ttngu207/main

update: DLC dependency from `pytorch_dlc` branch
datajoint · Feb 25, 2025 · 4acd1ee · 4acd1ee
2 parents 5f9edf1 + 102881d
commit 4acd1ee
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -24,30 +24,31 @@
     packages=find_packages(exclude=["contrib", "docs", "tests*"]),
     scripts=[],
     install_requires=[
-        "datajoint>=0.13.0",
+        "datajoint>=0.14.0",
         "graphviz",
         "pydot",
         "ipykernel",
         "ipywidgets",
     ],
     extras_require={
-        "dlc_default": ["deeplabcut[tf]>=2.2.1.1"],
+        "dlc_default": [
+            "deeplabcut[tf] @ git+/~https://github.com/DeepLabCut/DeepLabCut.git@pytorch_dlc"
+        ],
         "dlc_apple_mchips": [
             "tensorflow-macos==2.12.0",
             "tensorflow-metal",
             "tables==3.7.0",
             "deeplabcut",
         ],
-        "dlc_gui": ["deeplabcut[gui]"],
+        "dlc_gui": [
+            "deeplabcut[gui] @ git+/~https://github.com/DeepLabCut/DeepLabCut.git@pytorch_dlc"
+        ],
         "elements": [
             "element-lab @ git+/~https://github.com/datajoint/element-lab.git",
             "element-animal @ git+/~https://github.com/datajoint/element-animal.git",
             "element-session @ git+/~https://github.com/datajoint/element-session.git",
             "element-interface @ git+/~https://github.com/datajoint/element-interface.git",
         ],
         "tests": ["pytest", "pytest-cov", "shutils"],
-        "dlc-pytorch": [
-            "deeplabcut @ git+/~https://github.com/DeepLabCut/DeepLabCut.git@pytorch_dlc"
-        ],
     },
 )