Added pytdc to the tests

datamol-io · Jul 12, 2024 · 632d4dc · 632d4dc
1 parent b59428a
commit 632d4dc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -49,6 +49,9 @@ jobs:
       - name: Install library
         run: python -m pip install --no-deps -e . # `-e` required for correct `coverage` run.
 
+      - name: Install test dependencies
+        run: mamba install -c conda-forge pytdc  # Required to run the `test_finetuning.py`
+
       - name: Run tests
         run: pytest -m 'not ipu'
 

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -22,7 +22,6 @@
 import unittest as ut
 import gzip
 
-from graphium.utils.read_file import file_opener
 from graphium.utils.tensor import (
     nan_mad,
     nan_mean,