From 484f725fd36ba1d3700d4f9f064af34cb4b36cc4 Mon Sep 17 00:00:00 2001 From: Evgeni Burovski Date: Thu, 5 Dec 2024 18:44:56 +0200 Subject: [PATCH] MAINT: avoid a DeprecationWarning from torch --- vendor_test/uses_torch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vendor_test/uses_torch.py b/vendor_test/uses_torch.py index 5804aaff..747ecd51 100644 --- a/vendor_test/uses_torch.py +++ b/vendor_test/uses_torch.py @@ -23,7 +23,7 @@ def _test_torch(): assert isinstance(b, torch.Tensor) assert isinstance(res, torch.Tensor) - torch.testing.assert_allclose(res, [[1., 2., 3.]]) + torch.testing.assert_close(res, torch.as_tensor([[1., 2., 3.]])) assert is_torch_array(res) assert is_torch_namespace(torch) and is_torch_namespace(torch_compat)