add normal as default distribution

antonbaumann · May 2, 2024 · d334994 · d334994
1 parent 4356ae3
commit d334994
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utools/evaluation/regression.py b/utools/evaluation/regression.py
@@ -52,7 +52,7 @@ def regression_calibration_df(
     y_pred: torch.Tensor, 
     var_pred: torch.Tensor, 
     y_true: torch.Tensor,
-    distribution: Literal['normal', 'laplace'],
+    distribution: Literal['normal', 'laplace'] = 'normal',
     n_bins: int = 50,
     n_samples: Optional[int] = None,
     max_workers: int = 1,