Add mypy checks

casus · Sep 27, 2024 · 9ce7ca8 · 9ce7ca8
1 parent 74baf71
commit 9ce7ca8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -36,3 +36,8 @@ jobs:
     - name: Test with pytest
       run: |
         docker run --rm cvdm-package:${{ matrix.python-version }} pytest
+
+    # Step 5: Run the container and execute tests using mypy
+    - name: Test with mypy
+      run: |
+        docker run --rm cvdm-package:${{ matrix.python-version }} mypy .
diff --git a/cvdm/utils/inference_utils.py b/cvdm/utils/inference_utils.py
@@ -90,6 +90,7 @@ def create_output_montage(
             (pred_y, y, gamma_vec),
             axis=2,
         )
+    print(concatenated_images.shape)
     image: np.ndarray = montage(
         np.squeeze(concatenated_images),
         channel_axis=channel_axis,