bug fix

mpelchat04 · Nov 2, 2021 · 6180168 · 6180168
1 parent b37f61b
commit 6180168
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/airborne_lidar/airborne_lidar_seg.py b/airborne_lidar/airborne_lidar_seg.py
@@ -17,11 +17,10 @@
 import torch.nn.functional as F
 import convpoint.knn.lib.python.nearest_neighbors as nearest_neighbors
 import utils.metrics as metrics
-from airborne_lidar.airborne_lidar_utils import InformationLogger, print_metric, write_config, read_parameters, wblue, wgreen, write_features
+from airborne_lidar.airborne_lidar_utils import InformationLogger, print_metric, write_config, read_parameters, wblue, wgreen, write_features, write_to_las
 from pathlib import Path
 from airborne_lidar.airborne_lidar_datasets import PartDatasetTrainVal, PartDatasetTest
 from airborne_lidar.prepare_airborne_lidar_label import read_las_format
-from airborne_lidar.airborne_lidar_inference import write_to_las
 from mlflow import log_params, set_tracking_uri, set_experiment