Fix tfds installation (#25)

cvat-ai · May 10, 2023 · 9a97050 · 9a97050
1 parent 22401b0
commit 9a97050
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -74,8 +74,11 @@ def parse_requirements(filename=CORE_REQUIREMENTS_FILE):
     extras_require={
         "tf": ["tensorflow"],
         "tfds": [
-            "tensorflow-datasets!=4.5.0,!=4.5.1"
-        ],  # 4.5.0 fails on Windows, https://github.com/tensorflow/datasets/issues/3709
+            # 4.5.0 fails on Windows, https://github.com/tensorflow/datasets/issues/3709
+            "tensorflow-datasets!=4.5.0,!=4.5.1",
+            # TFDS misses a dependency in recent python releases
+            "dill~=0.3",
+        ],
         "tf-gpu": ["tensorflow-gpu"],
         "default": DEFAULT_REQUIREMENTS,
     },