traceback

gradio_app.py

@@ -187,8 +187,14 @@ def train_model(args: argparse.Namespace) -> None:
         training_state.model.optimize()
 
     except Exception as e:
+        import traceback
+
         training_state.results.training_logs.append(f"ERROR: {str(e)}")
+        training_state.results.training_logs.append(
+            f"TRACEBACK: {traceback.format_exc()}"
+        )
         logging.error(f"Training failed: {str(e)}")
+        logging.error(f"TRACEBACK: {traceback.format_exc()}")
     finally:
         training_state.is_training = False
 

gradio_models.py

@@ -388,7 +388,9 @@ class GradioGaussianSplatting2D(nn.Module):
             )
 
         begin = perf_counter()
-        tmp = project_gaussians_2d_scale_rot(xy, scale, rot, int(img_h), int(img_w), tile_bounds)
+        tmp = project_gaussians_2d_scale_rot(
+            xy, scale, rot, int(img_h), int(img_w), tile_bounds
+        )
         xy, radii, conics, num_tiles_hit = tmp
 
         if not self.disable_tiles:

pyproject.docker.toml

@@ -45,3 +45,4 @@ explicit = true
 dev = [
     "huggingface-hub[cli]>=0.34.4",
 ]
+

uv.lock

@@ -621,16 +621,36 @@ wheels = [
 [[package]]
 name = "gsplat"
 version = "0.1.0"
-source = { registry = "https://pypi.org/simple" }
+source = { directory = "gsplat" }
 dependencies = [
     { name = "jaxtyping" },
+    { name = "numpy", version = "2.2.6", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.11'" },
+    { name = "numpy", version = "2.3.3", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.11'" },
     { name = "rich" },
     { name = "torch", version = "2.6.0+cu124", source = { registry = "https://download.pytorch.org/whl/cu124" }, marker = "sys_platform == 'linux'" },
     { name = "torch", version = "2.8.0", source = { registry = "https://pypi.org/simple" }, marker = "sys_platform != 'linux'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/ef/df/9b1e8b5b0b5e604d967030f49cad7991a897291a8d84a07f7be987a85c34/gsplat-0.1.0.tar.gz", hash = "sha256:63defb3dddaacf10c03937a40b605cf25e0ae5befee219e792c02adec05f5b62", size = 4394498, upload-time = "2023-12-13T20:56:15.711Z" }
-wheels = [
-    { url = "https://files.pythonhosted.org/packages/32/25/1068751c23bd8201fdeed31d7b7e1c3be30cebb30323274a0c49b4a7c168/gsplat-0.1.0-py3-none-any.whl", hash = "sha256:2990e83a96a485920e5e34bd2cad11b7f445ece6199506d3d1d22f975d879c19", size = 5960717, upload-time = "2023-12-13T20:56:12.705Z" },
+
+[package.metadata]
+requires-dist = [
+    { name = "jaxtyping", specifier = ">=0.3.2" },
+    { name = "numpy", specifier = ">=2.2.6" },
+    { name = "rich", specifier = ">=14.1.0" },
+    { name = "torch", specifier = ">=2.6.0" },
+]
+
+[package.metadata.requires-dev]
+dev = [
+    { name = "black", specifier = ">=25.1.0" },
+    { name = "build", specifier = ">=1.3.0" },
+    { name = "isort", specifier = ">=6.0.1" },
+    { name = "ninja", specifier = ">=1.13.0" },
+    { name = "pylint", specifier = ">=3.3.8" },
+    { name = "pytest", specifier = ">=8.4.2" },
+    { name = "pytest-xdist", specifier = ">=3.8.0" },
+    { name = "pyyaml", specifier = ">=6.0.2" },
+    { name = "twine", specifier = ">=6.2.0" },
+    { name = "typeguard", specifier = ">=4.4.4" },
 ]
 
 [[package]]
@@ -752,7 +772,7 @@ dev = [
 requires-dist = [
     { name = "fused-ssim", git = "https://github.com/rahul-goel/fused-ssim/" },
     { name = "gradio", specifier = ">=4.0.0" },
-    { name = "gsplat" },
+    { name = "gsplat", directory = "gsplat" },
     { name = "huggingface-hub", specifier = ">=0.24.0" },
     { name = "lpips", specifier = ">=0.1.4" },
     { name = "matplotlib", specifier = ">=3.10.6" },