Fix

NicolasHug · NicolasHug · commit 5825706277e8 · 2025-06-03T13:36:39.000+01:00
diff --git a/test/test_tv_tensors.py b/test/test_tv_tensors.py
@@ -139,7 +139,8 @@ def test_to_wrapping(make_input):
 )
 @pytest.mark.parametrize("return_type", ["Tensor", "TVTensor"])
 def test_to_tv_tensor_reference(make_input, return_type):
-    tensor = torch.rand((3, 16, 16), dtype=torch.float64)
+    tensor = make_input().to(dtype=torch.float64).as_subclass(torch.Tensor)
+    assert type(tensor) is torch.Tensor
     dp = make_input()
 
     with tv_tensors.set_return_type(return_type):

Original file line number	Diff line number	Diff line change
`@@ -139,7 +139,8 @@ def test_to_wrapping(make_input):`
`139`	`139`	`)`
`140`	`140`	`@pytest.mark.parametrize("return_type", ["Tensor", "TVTensor"])`
`141`	`141`	`def test_to_tv_tensor_reference(make_input, return_type):`
`142`		`- tensor = torch.rand((3, 16, 16), dtype=torch.float64)`
	`142`	`+ tensor = make_input().to(dtype=torch.float64).as_subclass(torch.Tensor)`
	`143`	`+ assert type(tensor) is torch.Tensor`
`143`	`144`	`dp = make_input()`
`144`	`145`
`145`	`146`	`with tv_tensors.set_return_type(return_type):`