Remove mxnet and tensorflow-addons

Signed-off-by: Beat Buesser <[email protected]>

Author: beat-buesser

.github/workflows/ci-huggingface.yml

@@ -49,7 +49,7 @@ jobs:
           sudo apt-get update
           sudo apt-get -y -q install ffmpeg libavcodec-extra
           python -m pip install --upgrade pip setuptools wheel
-          pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^mxnet/d' requirements_test.txt)
+          pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^torch/d;/^torchvision/d;/^torchaudio/d;/^transformers/d' requirements_test.txt)
           pip install tensorflow==2.18.1
           pip install keras==3.10.0
           pip install torch==${{ matrix.torch }} --index-url https://download.pytorch.org/whl/cpu

.github/workflows/ci-keras.yml

@@ -48,7 +48,7 @@ jobs:
           sudo apt-get update
           sudo apt-get -y -q install ffmpeg libavcodec-extra
           python -m pip install --upgrade pip setuptools wheel
-          pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^tensorflow-addons/d;/^mxnet/d' requirements_test.txt)
+          pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^tensorflow-addons/d' requirements_test.txt)
           pip install tensorflow==${{ matrix.tensorflow }}
           pip install keras==${{ matrix.keras }}
           pip install tensorflow-addons==${{ matrix.tf_addons }}

.github/workflows/ci-legacy.yml

@@ -52,7 +52,7 @@ jobs:
           sudo apt-get update
           sudo apt-get -y -q install ffmpeg libavcodec-extra
           python -m pip install --upgrade pip setuptools wheel
-          pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^torch/d;/^torchvision/d;/^torchaudio/d;/^mxnet/d' requirements_test.txt)
+          pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^torch/d;/^torchvision/d;/^torchaudio/d' requirements_test.txt)
           pip install tensorflow==${{ matrix.tensorflow }}
           pip install keras==${{ matrix.keras }}
           pip install scikit-learn==${{ matrix.scikit-learn }}

.github/workflows/ci-pytorch-object-detectors.yml

@@ -39,7 +39,7 @@ jobs:
           sudo apt-get update
           sudo apt-get -y -q install ffmpeg libavcodec-extra
           python -m pip install --upgrade pip setuptools wheel
-          pip install -q -r <(sed '/^mxnet/d' requirements_test.txt)
+          pip install -q -r requirements_test.txt
           pip list
       - name: Run Test Action - test_pytorch_object_detector
         run: pytest --cov-report=xml --cov=art --cov-append -q -vv tests/estimators/object_detection/test_pytorch_object_detector.py --framework=pytorch --durations=0

.github/workflows/ci-pytorch.yml

@@ -54,7 +54,7 @@ jobs:
           sudo apt-get update
           sudo apt-get -y -q install ffmpeg libavcodec-extra
           python -m pip install --upgrade pip setuptools wheel
-          pip install -q -r <(sed '/^mxnet/d' requirements_test.txt)
+          pip install -q -r <(sed '/^torch/d;/^torchvision/d;/^torchaudio/d' requirements_test.txt)
           pip install torch==${{ matrix.torch }} --index-url https://download.pytorch.org/whl/cpu
           pip install torchvision==${{ matrix.torchvision }} --index-url https://download.pytorch.org/whl/cpu
           pip install torchaudio==${{ matrix.torchaudio }} --index-url https://download.pytorch.org/whl/cpu

.github/workflows/ci-style-checks.yml

@@ -39,7 +39,7 @@ jobs:
       - name: Install Dependencies
         run: |
           python -m pip install --upgrade pip setuptools wheel
-          pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^mxnet/d' requirements_test.txt)
+          pip install -q -r <(sed '/^tensorflow/d;/^keras/d' requirements_test.txt)
           pip install tensorflow==2.18.1
           pip install keras==3.10.0
           pip list

.github/workflows/ci-tensorflow-v2.yml

@@ -49,7 +49,7 @@ jobs:
           sudo apt-get update
           sudo apt-get -y -q install ffmpeg libavcodec-extra
           python -m pip install --upgrade pip setuptools wheel
-          pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^tensorflow-addons/d;/^mxnet/d' requirements_test.txt)
+          pip install -q -r <(sed '/^tensorflow/d;/^keras/d;/^tensorflow-addons/d' requirements_test.txt)
           pip install tensorflow==${{ matrix.tensorflow }}
           pip install keras==${{ matrix.keras }}
           pip install tensorflow-addons==${{ matrix.tf_addons }}

requirements_test.txt

@@ -26,7 +26,6 @@ multiprocess>=0.70.12
 
 tensorflow==2.18.1
 keras==3.10.0
-tensorflow-addons==0.23.0
 
 # PyTorch
 torch==2.7.0

setup.py

@@ -62,9 +62,9 @@ def get_version(rel_path):
         "gpy": ["GPy"],
         "keras": ["keras", "h5py"],
         "lightgbm": ["lightgbm"],
-        "tensorflow": ["tensorflow", "tensorflow_addons", "h5py"],
-        "tensorflow_image": ["tensorflow", "tensorflow_addons", "h5py", "Pillow", "ffmpeg-python", "opencv-python"],
-        "tensorflow_audio": ["tensorflow", "tensorflow_addons", "h5py", "pydub", "resampy", "librosa"],
+        "tensorflow": ["tensorflow", "h5py"],
+        "tensorflow_image": ["tensorflow", "h5py", "Pillow", "ffmpeg-python", "opencv-python"],
+        "tensorflow_audio": ["tensorflow", "h5py", "pydub", "resampy", "librosa"],
         "pytorch": ["torch", "torchvision"],
         "pytorch_image": ["torch", "torchvision", "kornia", "Pillow", "ffmpeg-python", "opencv-python"],
         "pytorch_audio": ["torch", "torchvision", "torchaudio", "pydub", "resampy", "librosa"],