From 9f07f3e36c3e9ff721423a7cfa165da7e61e7fa0 Mon Sep 17 00:00:00 2001 From: Niklas Netter Date: Fri, 11 Aug 2023 11:07:43 +0200 Subject: [PATCH] remove ignore statements --- tests/test_data_2d.py | 14 ++++---------- tests/test_model_2d.py | 4 ++-- tests/test_model_3d.py | 4 ++-- 3 files changed, 8 insertions(+), 14 deletions(-) diff --git a/tests/test_data_2d.py b/tests/test_data_2d.py index 1fa54a6..f7a5365 100644 --- a/tests/test_data_2d.py +++ b/tests/test_data_2d.py @@ -51,21 +51,15 @@ def test_getitem( slice_x1 = slice(i * n_channel, (i + 1) * n_channel) slice_x2 = slice((i + 1) * n_channel, (i + 2) * n_channel) - np.testing.assert_equal( # type: ignore [no-untyped-call] - new_x[..., slice_x1], new_x[..., slice_x2] - ) + np.testing.assert_equal(new_x[..., slice_x1], new_x[..., slice_x2]) - np.testing.assert_equal( # type: ignore [no-untyped-call] - probs[..., i], probs[..., i + 1] - ) + np.testing.assert_equal(probs[..., i], probs[..., i + 1]) slice_y1 = slice(i * n_rays, (i + 1) * n_rays) slice_y2 = slice((i + 1) * n_rays, (i + 2) * n_rays) - np.testing.assert_equal( # type: ignore [no-untyped-call] - dists[..., slice_y1], dists[..., slice_y2] - ) + np.testing.assert_equal(dists[..., slice_y1], dists[..., slice_y2]) - np.testing.assert_equal( # type: ignore [no-untyped-call] + np.testing.assert_equal( dists[..., mask_start + i], dists[..., mask_start + i + 1] ) diff --git a/tests/test_model_2d.py b/tests/test_model_2d.py index 1e78c82..6c9d5fc 100644 --- a/tests/test_model_2d.py +++ b/tests/test_model_2d.py @@ -23,8 +23,8 @@ def test_model_conf_train_predict( img = np.repeat(img[..., np.newaxis], n_channel, axis=-1) imgs = [ - np.copy(img), # type: ignore [no-untyped-call] - np.copy(img), # type: ignore [no-untyped-call] + np.copy(img), + np.copy(img), img, ] diff --git a/tests/test_model_3d.py b/tests/test_model_3d.py index 00d1cea..aefab04 100644 --- a/tests/test_model_3d.py +++ b/tests/test_model_3d.py @@ -23,8 +23,8 @@ def test_model_conf_train_predict( img = np.squeeze(img) imgs = [ - np.copy(img), # type: ignore [no-untyped-call] - np.copy(img), # type: ignore [no-untyped-call] + np.copy(img), + np.copy(img), img, ]