Merge pull request #4 from pomonam/documentation

Configure tests
pomonam · Mar 19, 2024 · 70e9696 · 70e9696
2 parents f37d027 + a11ee2f
commit 70e9696
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/.github/workflows/python-test.yml b/.github/workflows/python-test.yml
@@ -24,12 +24,8 @@ jobs:
     - name: Run All Tests
       run: |
         pip install -e ."[dev]"
-        pytest -vx tests/test_testable_tasks.py
         pytest -vx tests/test_analyzer.py
         pytest -vx tests/test_modules.py
-        pytest -vx tests/test_samplers.py
         pytest -vx tests/test_per_sample_gradients.py
         pytest -vx tests/factors/test_covariances.py
-        pytest -vx tests/factors/test_eigens.py
         pytest -vx tests/scores/test_pairwise_scores.py
-        pytest -vx tests/scores/test_self_scores.py
diff --git a/examples/uci/requirements.txt b/examples/uci/requirements.txt
@@ -0,0 +1,2 @@
+scikit-learn
+jupyter
diff --git a/tests/test_per_sample_gradients.py b/tests/test_per_sample_gradients.py
@@ -413,9 +413,9 @@ def test_query_gradient_svd(
 
 
 @pytest.mark.parametrize("input_dim", [256, 512])
-@pytest.mark.parametrize("output_dim", [512, 1024, 2048])
+@pytest.mark.parametrize("output_dim", [512, 1024])
 @pytest.mark.parametrize("batch_dim", [8, 16])
-@pytest.mark.parametrize("qbatch_dim", [8, 16, 128])
+@pytest.mark.parametrize("qbatch_dim", [8, 16])
 @pytest.mark.parametrize("rank", [32])
 @pytest.mark.parametrize("seed", [0])
 def test_query_gradient_svd_reconst(