type cast to int

flash3d/unidepth/utils/geometric.py

@@ -124,6 +124,8 @@ def unproject_points(
         torch.Tensor: Batch of 3D point clouds of shape (B, 3, H, W).
     """
     batch_size, _, height, width = depth.shape
+    height = int(height)
+    width = int(width)
     device = depth.device
 
     # Create pixel grid