fix test

pygod-team · Sep 20, 2024 · fa36957 · fa36957
1 parent 9f35955
commit fa36957
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.github/workflows/testing-cron.yml b/.github/workflows/testing-cron.yml
@@ -32,6 +32,7 @@ jobs:
         pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu
         pip install torch_geometric
         pip install torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.3.0+cpu.html
+        pip install scikit-learn
         pip install pytest
         pip install coverage
         pip install coveralls

diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml
@@ -37,6 +37,7 @@ jobs:
         pip install torch==2.3.0 --index-url https://download.pytorch.org/whl/cpu
         pip install torch_geometric
         pip install torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.3.0+cpu.html
+        pip install scikit-learn
         pip install pytest
         pip install coverage
         pip install coveralls