Fix type of input_dim in preprocessing params. It is now tuple, not list

Deci-AI · May 31, 2023 · 2c9ab37 · 2c9ab37
1 parent d978f91
commit 2c9ab37
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unit_tests/preprocessing_unit_test.py b/tests/unit_tests/preprocessing_unit_test.py
@@ -21,9 +21,9 @@ def test_getting_preprocessing_params(self):
             "ComposeProcessing": {
                 "processings": [
                     "ReverseImageChannels",
-                    {"DetectionLongestMaxSizeRescale": {"output_shape": [512, 512]}},
-                    {"DetectionLongestMaxSizeRescale": {"output_shape": [512, 512]}},
-                    {"DetectionBottomRightPadding": {"output_shape": [512, 512], "pad_value": 114}},
+                    {"DetectionLongestMaxSizeRescale": {"output_shape": (512, 512)}},
+                    {"DetectionLongestMaxSizeRescale": {"output_shape": (512, 512)}},
+                    {"DetectionBottomRightPadding": {"output_shape": (512, 512), "pad_value": 114}},
                     {"ImagePermute": {"permutation": (2, 0, 1)}},
                 ]
             }