fix: requirements and workflow

js-ish · Jul 26, 2024 · a1b10d4 · a1b10d4
1 parent 6887cef
commit a1b10d4
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -24,7 +24,7 @@ jobs:
           cache: 'pip'
       - run: pip install -U flake8
       - name: Run python code lint
-        run: flake8 moltx
+        run: flake8 dilipredict
 
   tests:
     name: "Python ${{ matrix.py-version }} unit-tests"

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,6 @@
-einops
+numpy==1.24.4
 torch==1.13.1
 torchvision==0.14.1
+einops
+pillow
+scikit-learn
diff --git a/tests/test_datasets.py b/tests/test_datasets.py
diff --git a/tests/test_piplines.py b/tests/test_piplines.py
@@ -1,7 +1,7 @@
 from dilipredict import models, pipelines
 
 
-def test_MutSmiReg(image_loader):
+def test_DILIPredict(image_loader):
     test_img_path = 'tests/data/test_img.tiff'
     image_files = [
         [test_img_path] * 10,