fix: point cloud losses optional dependencies

InterDigitalInc · Feb 2, 2024 · 3697824 · 3697824
1 parent a4ec525
commit 3697824
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/compressai/losses/pointcloud/chamfer.py b/compressai/losses/pointcloud/chamfer.py
@@ -33,7 +33,11 @@
 import torch.nn as nn
 
 from einops import rearrange
-from pointops.functions import pointops
+
+try:
+    from pointops.functions import pointops
+except ImportError:
+    pass  # NOTE: Optional dependency.
 
 from compressai.layers.pointcloud.hrtzxf2022 import index_points
 from compressai.losses.utils import compute_rate_loss