Add type checking

.github/workflows/code_checks.yml

@@ -34,4 +34,4 @@ jobs:
         pylint mattport --max-line-length=120 --generated-members=numpy.*,torch.*,cv2.*,cv.* --good-names=e,c,i,j,x,y,lr,K,fx,fy,cx,cy,H,W --disable=too-many-arguments --disable=fixme
     - name: Test with pytest
       run: |
-        pytest tests --ignore tests/nerf/test_ray_generator.py
+        pytest tests --ignore tests/nerf/test_ray_generator.py --typeguard-packages=mattport --torchtyping-patch-typeguard