fix

inference.py

@@ -102,7 +102,7 @@ class DepthPredictor:
                 mask = mask.ravel()
             else:
                 mask = mask.ravel()[:-1]
-                
+            mask = mask.ravel()
             point_cloud_subset, color_array_subset = point_cloud[mask], color_array[mask]
             idxs = np.random.choice(len(point_cloud_subset), int(n_samples))
             point_cloud_subset = point_cloud_subset[idxs]