fix: config_data['true_depth'] cast to float

dattalab · Dec 30, 2021 · 3f7c582 · 3f7c582
1 parent 3a2233d
commit 3f7c582
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/moseq2_extract/helpers/wrappers.py b/moseq2_extract/helpers/wrappers.py
@@ -397,8 +397,8 @@ def extract_wrapper(input_file, output_dir, config_data, num_frames=None, skip=F
 
     status_dict['complete'] = True
     if status_dict['parameters'].get('true_depth') is None:
-        status_dict['parameters']['true_depth'] = config_data.get('true_depth')
-
+        # 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)