chore: update internal for CI

ZettaAI · May 7, 2024 · 8ffcb3c · 8ffcb3c
1 parent 0471b88
commit 8ffcb3c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/unit/augmentations/test_misalign.py b/tests/unit/augmentations/test_misalign.py
@@ -171,7 +171,7 @@ def test_dict_process_step_no_keys():
     )
     proc.prepared_disp_fraction = (1 / 5, 1 / 5)
     result = proc.process_data(data, mode="read")
-    assert result["key"].shape == [1, 4, 4, 5]
+    assert list(result["key"].shape) == [1, 4, 4, 5]
 
 
 @pytest.mark.parametrize(

diff --git a/zetta_utils/internal b/zetta_utils/internal