add function to retrieve a normalization by str

mu_map/dataset/normalization.py

@@ -43,6 +43,23 @@ class GaussianNormTransform(Transform):
         return norm_gaussian(inputs), outputs_expected
 
 
+norm_choices = ["max", "mean", "gaussian"]
+
+
+def norm_by_str(norm: str):
+    if norm is None:
+        return None
+
+    if norm == "mean":
+        return MeanNormTransform()
+    elif norm == "max":
+        return MaxNormTransform()
+    elif norm == "gaussian":
+        return GaussianNormTransform()
+
+    raise ValueError(f"Unknown normalization {norm}")
+
+
 __all__ = [
     norm_max.__name__,
     norm_mean.__name__,