Make kaolin optional for FlexiCubes import on Spaces

trellis/representations/mesh/flexicubes/flexicubes.py

@@ -8,7 +8,21 @@
 
 import torch
 from .tables import *
-from kaolin.utils.testing import check_tensor
+
+try:
+    from kaolin.utils.testing import check_tensor
+except ImportError:
+    def check_tensor(tensor, shape, throw=False):
+        """Fallback for kaolin.utils.testing.check_tensor used in Spaces builds."""
+        ok = torch.is_tensor(tensor) and tensor.ndim == len(shape)
+        if ok:
+            for i, expected in enumerate(shape):
+                if expected is not None and tensor.shape[i] != expected:
+                    ok = False
+                    break
+        if throw and not ok:
+            raise ValueError(f"Tensor shape {tuple(tensor.shape)} does not match expected {shape}")
+        return ok
 
 __all__ = [
     'FlexiCubes'