feat: ensure true_depth is added to result_00.yaml

dattalab · Dec 30, 2021 · 7745e0d · 7745e0d
2 parents be02f1e + 3f7c582
commit 7745e0d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/moseq2_extract/helpers/wrappers.py b/moseq2_extract/helpers/wrappers.py
@@ -396,7 +396,9 @@ def extract_wrapper(input_file, output_dir, config_data, num_frames=None, skip=F
         print(e)
 
     status_dict['complete'] = True
-
+    if status_dict['parameters'].get('true_depth') is None:
+        # config_data.get('true_depth') is numpy.float64 and yaml.safe_dump can't represent the object
+        status_dict['parameters']['true_depth'] = float(config_data.get('true_depth'))
     with open(status_filename, 'w') as f:
         yaml.safe_dump(status_dict, f)