Fix Open3D import

agml/data/point_cloud.py

@@ -21,14 +21,6 @@
 from agml.utils.general import is_float, is_int
 from agml.utils.logging import log
 
-try:
-    import open3d as o3d
-except:
-    log("Open3D is not installed. If you want to have high-res point cloud "
-        "visualizations, please install Open3D using `pip install open3d`. "
-        "For now, defaulting to using 3D matplotlib visualizations.")
-    o3d = None
-
 
 class PointCloud(object):
     """Represents a 3D point cloud object, with utilities for format + visualization."""
@@ -183,6 +175,14 @@ def _read_point_cloud(self, contents):
     def _build_3d_object(self):
         """Function to construct the point cloud as a open3d point cloud object
         """
+        try:
+            import open3d as o3d
+        except:
+            log("Open3D is not installed. If you want to have high-res point cloud "
+                "visualizations, please install Open3D using `pip install open3d`. "
+                "For now, defaulting to using 3D matplotlib visualizations.")
+            o3d = None
+
         # Construct the point cloud with the points and colors.
         if o3d is not None:
             self._structure_3d = o3d.geometry.PointCloud()

agml/viz/point_clouds.py

@@ -16,7 +16,8 @@
 
 import matplotlib.pyplot as plt
 
-from agml.data.point_cloud import PointCloud, o3d
+from agml.data.point_cloud import PointCloud
+from agml.utils.logging import log
 
 
 def show_point_cloud(point_cloud, format = 'default'):
@@ -33,6 +34,14 @@ def show_point_cloud(point_cloud, format = 'default'):
         done using Open3D. If 'matplotlib', then the visualization will be
         done using matplotlib. Defaults to 'default'.
     """
+    try:
+        import open3d as o3d
+    except:
+        log("Open3D is not installed. If you want to have high-res point cloud "
+            "visualizations, please install Open3D using `pip install open3d`. "
+            "For now, defaulting to using 3D matplotlib visualizations.")
+        o3d = None
+
     # We need to hard-code this check because this method isn't designed to
     # run with regular point cloud arrays, only `point_cloud` objects.
     if not isinstance(point_cloud, PointCloud):

requirements.txt

@@ -2,6 +2,7 @@ numpy
 matplotlib
 tqdm
 scikit-learn
+requests>=2.0.0
 opencv-python; sys.platform != 'linux'
 opencv-python-headless; sys.platform == 'linux'
 pyyaml>=5.4.1