Add mypy checks

casus · Sep 27, 2024 · 91590ed · 91590ed
1 parent 74baf71
commit 91590ed
Show file tree

Hide file tree

Showing 3 changed files with 7 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/Dockerfile b/Dockerfile
@@ -49,6 +49,7 @@ RUN micromamba create -n $ENV_NAME -y && \
     /usr/local/bin/micromamba run -n $ENV_NAME pip3 install opencv-python==4.9.0.80 \
     tensorflow-addons==0.23.0 \
     cupy-cuda12x==13.3.0 \
+    mypy \
     pytest && \
     /usr/local/bin/micromamba run -n $ENV_NAME python -m pip install .
 

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,