From a11ee2fed059ec2a28453a62f7a06601f6226b14 Mon Sep 17 00:00:00 2001 From: Juhan Bae Date: Tue, 19 Mar 2024 04:34:47 -0400 Subject: [PATCH] Configure tests --- .github/workflows/python-test.yml | 4 ---- examples/uci/requirements.txt | 2 ++ tests/test_per_sample_gradients.py | 4 ++-- 3 files changed, 4 insertions(+), 6 deletions(-) diff --git a/.github/workflows/python-test.yml b/.github/workflows/python-test.yml index ec59baa..38cde6f 100644 --- 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 index e69de29..4a67121 100644 --- a/examples/uci/requirements.txt +++ b/examples/uci/requirements.txt @@ -0,0 +1,2 @@ +scikit-learn +jupyter \ No newline at end of file diff --git a/tests/test_per_sample_gradients.py b/tests/test_per_sample_gradients.py index 5de7eb5..a5d333a 100644 --- 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(