Fix tests

Author: MartinBraquet

.gitignore

@@ -14,7 +14,6 @@ src/gcaa/.ipynb_checkpoints/
 */.ipynb_checkpoints/*
 */.idea/*
 */.pytest_cache/*
-/src/gcaa/torchlogs/*
 *.coverage
 *coverage.xml
 report

gcaa/tools/basic.py

@@ -1,8 +1,6 @@
 from collections import defaultdict
 
 import numpy as np
-import torch
-
 
 class PrettyDict(dict):
     def __repr__(self):
@@ -15,22 +13,15 @@ def dict_factory():
 
 def check_nan(data):
     """
-    >>> check_nan(torch.Tensor([1, 2, 3]))
-    >>> check_nan(np.array([1, 2, 3]))
     >>> check_nan(None)
-    >>> check_nan(torch.Tensor([1, float('nan'), 3]))
-    Traceback (most recent call last):
-       ...
-    RuntimeError: NaN detected in: tensor([1., nan, 3.])
+    >>> check_nan(np.array([1, 2, 3]))
     >>> check_nan(np.array([1, float('nan'), 3]))
     Traceback (most recent call last):
        ...
     RuntimeError: NaN detected in: [ 1. nan  3.]
     """
     if data is None:
         return
-    if isinstance(data, torch.Tensor) and not torch.isnan(data).any():
-        return
     if isinstance(data, np.ndarray) and not np.isnan(data).any():
         return
     raise RuntimeError(f"NaN detected in: {data}")

scripts/test_mypi.sh

@@ -2,7 +2,6 @@
 
 conda create -n test python=3.12 pip -y
 conda activate test
-pip install torch --index-url https://download.pytorch.org/whl/cpu
 pip install gcaa
 python -c "from gcaa import GCAA; GCAA().run()"
 echo "Installation from Pypi ran a test successfully"