From 83c7a690d741bdc5ce88af33678b6676afa9264b Mon Sep 17 00:00:00 2001
From: genisplaja <genis.plaja@upf.edu>
Date: Tue, 5 Nov 2024 17:34:49 +0100
Subject: [PATCH] add all tests

---
 .github/environment-ci.yml |  8 ++++++--
 .github/workflows/ci.yml   | 20 ++++++++++++++++++--
 2 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/.github/environment-ci.yml b/.github/environment-ci.yml
index 18577c3f..cf56091e 100644
--- a/.github/environment-ci.yml
+++ b/.github/environment-ci.yml
@@ -1,4 +1,4 @@
-name: mirdata-dev
+name: compiam-dev
 channels:
   - conda-forge
   - defaults
@@ -31,8 +31,12 @@ dependencies:
   - libvorbis
   # optional, but required for testing
   - pytest>=7.4.3
-
   - pip:
+    # Optional deps
+    - "tensorflow>=2.12.0"
+    - "tensorflow_addons"
+    - "torch==1.13.0"
+    - "essentia"
     - "soundfile>=0.12.1"
     - "opencv-python~=4.6.0"
     - "mirdata==0.3.8"
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 19e60ce7..449f20ed 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -103,6 +103,22 @@ jobs:
             conda info -a
             conda list
 
-        - name: Run pytest
+        - name: Run basic tests
           shell: bash -l {0}
-          run: pytest tests/
\ No newline at end of file
+          run: pytest tests/
+
+        - name: Run tensorflow tests
+          shell: bash -l {0}
+          run: pytest tests/ --tensorflow
+
+        - name: Run torch tests
+          shell: bash -l {0}
+          run: pytest tests/ --torch
+
+        - name: Run full ML tests
+          shell: bash -l {0}
+          run: pytest tests/ --full-ml
+
+        - name: Run all tests
+          shell: bash -l {0}
+          run: pytest tests/ --all
\ No newline at end of file