summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAyan Das <bvits@riseup.net>2025-06-08 20:52:41 +0530
committerLudovic Courtès <ludo@gnu.org>2025-06-13 12:22:40 +0200
commit1f1b65ddbca580c3b9893cbc01daa0de709d1d20 (patch)
treec7cb23315ad71388fbd14af9d363200e0061df92
parentf058c9e825fc19215cfd40b9ea6bb1817d735374 (diff)
gnu: python-captum: Fix tests for PyTorch 2.7.0.
* gnu/packages/machine-learning.scm (python-captum)[arguments]: Skip test_exp_sets_with_diffent_lengths which fails with PyTorch 2.7.0's stricter torch.load weights_only behavior. Signed-off-by: Ayan Das <bvits@riseup.net> Signed-off-by: Ludovic Courtès <ludo@gnu.org>
-rw-r--r--gnu/packages/machine-learning.scm13
1 files changed, 8 insertions, 5 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm
index 9636e2cd19..c1621d1e0b 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -5611,11 +5611,14 @@ Actions for the Lightning suite of libraries.")
(arguments
(list
#:test-flags
- '(list "-k"
- ;; These two tests (out of more than 1000 tests) fail because of
- ;; accuracy problems.
- "not test_softmax_classification_batch_multi_target\
- and not test_softmax_classification_batch_zero_baseline")))
+ '(list "-k" (string-append
+ ;; These two tests (out of more than 1000 tests) fail because of
+ ;; accuracy problems.
+ "not test_softmax_classification_batch_multi_target"
+ " and not test_softmax_classification_batch_zero_baseline"
+ ;; This test fails with PyTorch 2.7.0 due to stricter
+ ;; torch.load weights_only behavior.
+ " and not test_exp_sets_with_diffent_lengths"))))
(propagated-inputs
(list python-matplotlib python-numpy python-pytorch python-tqdm))
(native-inputs (list jupyter