test(audio): only install torchcodec when required

Author: dacorvo

.github/workflows/test_inf2_transformers.yml

@@ -56,6 +56,10 @@ jobs:
         uses: ./.github/actions/prepare_venv
       - name: Install optimum-neuron
         uses: ./.github/actions/install_optimum_neuron
+      - name: Install datasets dependencies
+        run: |
+          source aws_neuron_venv_pytorch/bin/activate
+          pip install datasets[audio]
       - name: Run transformers export CLI tests
         run: |
           source aws_neuron_venv_pytorch/bin/activate

pyproject.toml

@@ -68,7 +68,6 @@ tests = [
     "librosa",
     "controlnet-aux",
     "hf_transfer",
-    "torchcodec < 0.6.0",
 ]
 quality = [
     "pre-commit",