yaml config fixup

mpelchat04 · Jul 24, 2020 · 0fedb6d · 0fedb6d
1 parent 56d43e4
commit 0fedb6d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/airborne_lidar/airborne_lidar_seg.py b/examples/airborne_lidar/airborne_lidar_seg.py
@@ -352,7 +352,7 @@ def test(args, filename, model_folder, info_class):
         # write error file.
         # error2ply(model_folder / f"{filename}_error.ply", xyz=xyz, labels=lbl, prediction=scores, info_class=info_class['class_info'])
 
-    if args['global']['savepts']:
+    if args['test']['savepts']:
         # Save predictions
         out_folder = model_folder / 'tst'
         out_folder.mkdir(exist_ok=True)