Skip to content

Commit 4443705

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 20bbf6b commit 4443705

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/transforms/test_generate_heatmap.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -227,16 +227,16 @@ def test_truncated_parameter(self):
227227
# Test that truncated parameter correctly controls window size
228228
pt = np.array([[8.0, 8.0]], dtype=np.float32)
229229
sigma = 2.0
230-
230+
231231
# Test with different truncated values
232232
small_truncated = GenerateHeatmap(sigma=sigma, spatial_shape=(32, 32), truncated=2.0)(pt)[0]
233233
default_truncated = GenerateHeatmap(sigma=sigma, spatial_shape=(32, 32), truncated=4.0)(pt)[0] # default
234234
large_truncated = GenerateHeatmap(sigma=sigma, spatial_shape=(32, 32), truncated=6.0)(pt)[0]
235-
235+
236236
# Larger truncated should capture more of the gaussian, resulting in slightly higher total sum
237237
self.assertLess(small_truncated.sum(), default_truncated.sum())
238238
self.assertLess(default_truncated.sum(), large_truncated.sum())
239-
239+
240240
# All should have same peak value (normalized to 1.0)
241241
np.testing.assert_allclose(small_truncated.max(), 1.0, rtol=1e-5)
242242
np.testing.assert_allclose(default_truncated.max(), 1.0, rtol=1e-5)

0 commit comments

Comments
 (0)