Skip to content

Commit 06bdc7b

Browse files
committed
fix test for pytorch 2.6
1 parent 767c68a commit 06bdc7b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_priors.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def test_multiple_priors(dtype):
186186
with tempfile.NamedTemporaryFile() as f:
187187
torch.save(model, f)
188188
f.seek(0)
189-
model2 = torch.load(f)
189+
model2 = torch.load(f, weights_only=False)
190190
priors2 = model2.model.prior_model
191191
assert len(priors2) == 3
192192
assert isinstance(priors2[0], ZBL)

0 commit comments

Comments
 (0)