Skip to content

Commit 4acd1ee

Browse files
authored
Merge pull request #136 from ttngu207/main
update: DLC dependency from `pytorch_dlc` branch
2 parents 5f9edf1 + 102881d commit 4acd1ee

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

setup.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -24,30 +24,31 @@
2424
packages=find_packages(exclude=["contrib", "docs", "tests*"]),
2525
scripts=[],
2626
install_requires=[
27-
"datajoint>=0.13.0",
27+
"datajoint>=0.14.0",
2828
"graphviz",
2929
"pydot",
3030
"ipykernel",
3131
"ipywidgets",
3232
],
3333
extras_require={
34-
"dlc_default": ["deeplabcut[tf]>=2.2.1.1"],
34+
"dlc_default": [
35+
"deeplabcut[tf] @ git+https://github.com/DeepLabCut/DeepLabCut.git@pytorch_dlc"
36+
],
3537
"dlc_apple_mchips": [
3638
"tensorflow-macos==2.12.0",
3739
"tensorflow-metal",
3840
"tables==3.7.0",
3941
"deeplabcut",
4042
],
41-
"dlc_gui": ["deeplabcut[gui]"],
43+
"dlc_gui": [
44+
"deeplabcut[gui] @ git+https://github.com/DeepLabCut/DeepLabCut.git@pytorch_dlc"
45+
],
4246
"elements": [
4347
"element-lab @ git+https://github.com/datajoint/element-lab.git",
4448
"element-animal @ git+https://github.com/datajoint/element-animal.git",
4549
"element-session @ git+https://github.com/datajoint/element-session.git",
4650
"element-interface @ git+https://github.com/datajoint/element-interface.git",
4751
],
4852
"tests": ["pytest", "pytest-cov", "shutils"],
49-
"dlc-pytorch": [
50-
"deeplabcut @ git+https://github.com/DeepLabCut/DeepLabCut.git@pytorch_dlc"
51-
],
5253
},
5354
)

0 commit comments

Comments
 (0)