diff --git a/pyproject.toml b/pyproject.toml
index 51cb67aa..44d52f22 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,7 +7,7 @@ dependencies = [
     "torch",
     "torchvision",
     "timm",
-    "numpy<2",
+    "numpy",
     "pillow_heif",
     "matplotlib",
 ]
@@ -20,7 +20,7 @@ Homepage = "https://github.com/apple/ml-depth-pro"
 Repository = "https://github.com/apple/ml-depth-pro"
 
 [build-system]
-requires = ["setuptools", "setuptools-scm"]
+requires = ["setuptools"]
 build-backend = "setuptools.build_meta"
 
 [tool.setuptools.packages.find]
diff --git a/src/depth_pro/depth_pro.py b/src/depth_pro/depth_pro.py
index f31b4e16..c0befa48 100644
--- a/src/depth_pro/depth_pro.py
+++ b/src/depth_pro/depth_pro.py
@@ -283,7 +283,6 @@ def infer(
             f_px = 0.5 * W / torch.tan(0.5 * torch.deg2rad(fov_deg.to(torch.float)))
         
         inverse_depth = canonical_inverse_depth * (W / f_px)
-        f_px = f_px.squeeze()
 
         if resize:
             inverse_depth = nn.functional.interpolate(