From b397daff36d3ede5e40650f057f16affc0af2244 Mon Sep 17 00:00:00 2001 From: Claudio Fantacci Date: Mon, 24 Aug 2026 02:54:22 -0700 Subject: [PATCH] Improve `test_pad_to_size_chw_values` test This is mostly to make static type analyzers happy. PiperOrigin-RevId: 969728255 --- dm_pix/_src/augment_test.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/dm_pix/_src/augment_test.py b/dm_pix/_src/augment_test.py index d41ffb2..7554157 100644 --- a/dm_pix/_src/augment_test.py +++ b/dm_pix/_src/augment_test.py @@ -590,6 +590,8 @@ def test_pad_to_size_chw_values(self): result = augment.pad_to_size( image, target_height=3, target_width=3, channel_axis=0 ) + self.assertNotIsInstance(result, np.bool_) + self.assertNotIsInstance(result, np.number) self.assertEqual(result.shape, (1, 3, 3)) np.testing.assert_array_equal(result[0, 1, 1], 1.0) np.testing.assert_array_equal(result[0, 0, :], 0.0)