[Fix] Fix incompatibility of ImgAug and latest Numpy (#2451)

open-mmlab · May 6, 2023 · 718a17d · 718a17d
1 parent 588acac
commit 718a17d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.circleci/test.yml b/.circleci/test.yml
@@ -114,7 +114,6 @@ jobs:
             docker build .circleci/docker -t mmaction:gpu --build-arg PYTORCH=<< parameters.torch >> --build-arg CUDA=<< parameters.cuda >> --build-arg CUDNN=<< parameters.cudnn >>
             docker run --gpus all -t -d -v /home/circleci/project:/mmaction -w /mmaction --name mmaction mmaction:gpu
             docker exec mmaction apt-get update
-            docker exec mmaction pip install "numpy==1.23"
             docker exec mmaction apt-get upgrade -y
             docker exec mmaction apt-get install -y ffmpeg libsm6 libxext6 git ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 libturbojpeg pkg-config
             docker exec mmaction apt-get install -y libavdevice-dev libavfilter-dev libopus-dev libvpx-dev libsrtp2-dev libsndfile1

diff --git a/mmaction/datasets/transforms/wrappers.py b/mmaction/datasets/transforms/wrappers.py
@@ -204,6 +204,9 @@ class ImgAug(BaseTransform):
     """
 
     def __init__(self, transforms):
+        # Hack to fix incompatibility of ImgAug and latest Numpy
+        if digit_version(np.__version__) >= digit_version('1.24.0'):
+            np.bool = bool
         import imgaug.augmenters as iaa
 
         if transforms == 'default':

diff --git a/src/pytorch-sphinx-theme b/src/pytorch-sphinx-theme