Revert "Fix"

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

Original file line number	Diff line number	Diff line change
`@@ -139,8 +139,7 @@ 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 = make_input().to(dtype=torch.float64).as_subclass(torch.Tensor)`
`143`		`- assert type(tensor) is torch.Tensor`
	`142`	`+ tensor = torch.rand((3, 16, 16), dtype=torch.float64)`
`144`	`143`	`dp = make_input()`
`145`	`144`
`146`	`145`	`with tv_tensors.set_return_type(return_type):`