ruff

app.py

@@ -21,7 +21,7 @@ import imageio
 import numpy as np
 import spaces
 import torch
-from easydict import EasyDict as edict
+from easydict import EasyDict
 from PIL import Image
 
 from trellis.pipelines import TrellisImageTo3DPipeline
@@ -90,7 +90,7 @@ def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
     }
 
 
-def unpack_state(state: dict) -> tuple[Gaussian, edict, str]:
+def unpack_state(state: dict) -> tuple[Gaussian, EasyDict, str]:
     gs = Gaussian(
         aabb=state["gaussian"]["aabb"],
         sh_degree=state["gaussian"]["sh_degree"],
@@ -105,7 +105,7 @@ def unpack_state(state: dict) -> tuple[Gaussian, edict, str]:
     gs._rotation = torch.tensor(state["gaussian"]["_rotation"], device="cuda")
     gs._opacity = torch.tensor(state["gaussian"]["_opacity"], device="cuda")
 
-    mesh = edict(
+    mesh = EasyDict(
         vertices=torch.tensor(state["mesh"]["vertices"], device="cuda"),
         faces=torch.tensor(state["mesh"]["faces"], device="cuda"),
     )