fix test error

eclipse0922 · eclipse0922 · commit eafe59a57eb9 · 2025-09-21T23:40:20.000+09:00
Signed-off-by: sewon.jeon &lt;sewon.jeon@connecteve.com&gt;
diff --git a/tests/transforms/test_generate_heatmap.py b/tests/transforms/test_generate_heatmap.py
@@ -55,14 +55,14 @@ def _argmax_nd(x) -> np.ndarray:
 
 # Test cases for 3D numpy outputs with explicit dtype
 TEST_CASES_3D_NUMPY = []
-for dtype in [np.float32, np.float64]:
+for dtype_obj in [np.float32, np.float64]:
     TEST_CASES_3D_NUMPY.append(
         [
-            f"3d_numpy_{dtype.__name__}",
+            f"3d_numpy_{dtype_obj.__name__}",
             np.array([[1.5, 2.5, 3.5]], dtype=np.float32),
-            {"sigma": 1.0, "spatial_shape": (8, 8, 8), "dtype": dtype},
+            {"sigma": 1.0, "spatial_shape": (8, 8, 8), "dtype": dtype_obj},
             (1, 8, 8, 8),
-            dtype,
+            dtype_obj,
         ]
     )
 

Original file line number	Diff line number	Diff line change
`@@ -55,14 +55,14 @@ def _argmax_nd(x) -> np.ndarray:`
`55`	`55`
`56`	`56`	`# Test cases for 3D numpy outputs with explicit dtype`
`57`	`57`	`TEST_CASES_3D_NUMPY = []`
`58`		`-for dtype in [np.float32, np.float64]:`
	`58`	`+for dtype_obj in [np.float32, np.float64]:`
`59`	`59`	`TEST_CASES_3D_NUMPY.append(`
`60`	`60`	`[`
`61`		`- f"3d_numpy_{dtype.__name__}",`
	`61`	`+ f"3d_numpy_{dtype_obj.__name__}",`
`62`	`62`	`np.array([[1.5, 2.5, 3.5]], dtype=np.float32),`
`63`		`- {"sigma": 1.0, "spatial_shape": (8, 8, 8), "dtype": dtype},`
	`63`	`+ {"sigma": 1.0, "spatial_shape": (8, 8, 8), "dtype": dtype_obj},`
`64`	`64`	`(1, 8, 8, 8),`
`65`		`- dtype,`
	`65`	`+ dtype_obj,`
`66`	`66`	`]`
`67`	`67`	`)`
`68`	`68`