Fixed failing GPU test.

miguelalonsojr · miguelalonsojr · commit 9d34978a4396 · 2024-10-04T15:58:03.000-04:00
diff --git a/ml-agents/mlagents/trainers/tests/test_torch_utils.py b/ml-agents/mlagents/trainers/tests/test_torch_utils.py
@@ -11,8 +11,8 @@
     "device_str, expected_type, expected_index, expected_tensor_type",
     [
         ("cpu", "cpu", None, torch.float32),
-        ("cuda", "cuda", None, torch.cuda.FloatTensor),
-        ("cuda:42", "cuda", 42, torch.cuda.FloatTensor),
+        ("cuda", "cuda", None, torch.float32),
+        ("cuda:42", "cuda", 42, torch.float32),
         ("opengl", "opengl", None, torch.float32),
     ],
 )

Original file line number	Diff line number	Diff line change
`@@ -11,8 +11,8 @@`
`11`	`11`	`"device_str, expected_type, expected_index, expected_tensor_type",`
`12`	`12`	`[`
`13`	`13`	`("cpu", "cpu", None, torch.float32),`
`14`		`- ("cuda", "cuda", None, torch.cuda.FloatTensor),`
`15`		`- ("cuda:42", "cuda", 42, torch.cuda.FloatTensor),`
	`14`	`+ ("cuda", "cuda", None, torch.float32),`
	`15`	`+ ("cuda:42", "cuda", 42, torch.float32),`
`16`	`16`	`("opengl", "opengl", None, torch.float32),`
`17`	`17`	`],`
`18`	`18`	`)`