revert trying to test mamba on workflow, since workflows do not have a GPU anyway

Author: LarsKue

.github/workflows/tests.yaml

@@ -61,13 +61,6 @@ jobs:
         run: |
           pip install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
 
-      # we currently have extra dependencies for torch tests
-      # TODO: can we install this dynamically? should we add it to the test dependencies?
-      - name: Install Extra Dependencies
-        if: ${{ matrix.backend == 'torch' }}
-        run: |
-          pip install -U mamba-ssm
-
       - name: Show Environment Info
         run: |
           python --version

tests/test_wrappers/test_mamba.py

@@ -5,13 +5,23 @@
 
 @pytest.mark.torch
 def test_mamba_summary(random_time_series, mamba_summary_network):
+    import torch
+
+    if not torch.cuda.is_available():
+        pytest.skip("This test requires a GPU environment.")
+
     out = mamba_summary_network(random_time_series)
     # Batch size 2, summary dim 4
     assert out.shape == (2, 4)
 
 
 @pytest.mark.torch
 def test_mamba_trains(random_time_series, inference_network, mamba_summary_network):
+    import torch
+
+    if not torch.cuda.is_available():
+        pytest.skip("This test requires a GPU environment.")
+
     workflow = bf.BasicWorkflow(
         inference_network=inference_network,
         summary_network=mamba_summary_network,