Fix numpy conflict (cvat-ai#278)

TOsmanov · Jun 3, 2021 · 723c500 · 723c500
1 parent bf32c4d
commit 723c500
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -6,7 +6,7 @@ lxml>=4.4.1
 matplotlib>=3.3.1
 opencv-python-headless>=4.1.0.25
 Pillow>=6.1.0
-pycocotools>=2.0.0
+pycocotools>=2.0.0 --no-binary=pycocotools # https://github.com/openvinotoolkit/datumaro/issues/253
 PyYAML>=5.3.1
 scikit-image>=0.15.0
 tensorboardX>=1.8

diff --git a/setup.py b/setup.py
@@ -42,8 +42,17 @@ def get_requirements():
         'matplotlib',
         'numpy>=1.17.3',
         'Pillow',
-        'pycocotools; platform_system != "Windows"',
+
+        # Avoid 2.0.2 Linux binary distribution because of
+        # a conflict in numpy versions with TensorFlow:
+        # - TF is compiled with numpy 1.19 ABI
+        # - pycocotools is compiled with numpy 1.20 ABI
+        # Using a previous version allows to force package rebuilding.
+        #
+        # https://github.com/openvinotoolkit/datumaro/issues/253
+        'pycocotools!=2.0.2; platform_system != "Windows"',
         'pycocotools-windows; platform_system == "Windows"',
+
         'PyYAML',
         'scikit-image',
         'tensorboardX',