diff --git a/moseq2_extract/helpers/wrappers.py b/moseq2_extract/helpers/wrappers.py index f5f4bd8..8ec3ffc 100644 --- a/moseq2_extract/helpers/wrappers.py +++ b/moseq2_extract/helpers/wrappers.py @@ -397,7 +397,7 @@ 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'] = int(config_data.get('true_depth')) + status_dict['parameters']['true_depth'] = config_data.get('true_depth') with open(status_filename, 'w') as f: yaml.safe_dump(status_dict, f)