round pi_times to 2 decimals

wasserth · Apr 9, 2024 · 3275f79 · 3275f79
1 parent 0cb65bd
commit 3275f79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/totalsegmentator/bin/totalseg_get_phase.py b/totalsegmentator/bin/totalseg_get_phase.py
@@ -84,7 +84,7 @@ def get_ct_contrast_phase(ct_img: nib.Nifti1Image):
     phase, probability = pi_time_to_phase(pi_time)
 
     return {"pi_time": pi_time, "phase": phase, "probability": probability, 
-            "pi_time_min": float(preds.min()), "pi_time_max": float(preds.max())}
+            "pi_time_min": round(float(preds.min()), 2), "pi_time_max": round(float(preds.max()), 2)}
 
 
 def main():