fix vector inputs

redmod-team · May 10, 2023 · ab6e84f · ab6e84f
1 parent 884f430
commit ab6e84f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/profit/util/util.py b/profit/util/util.py
@@ -54,7 +54,10 @@ def params2map(params: Union[None, MutableMapping, np.ndarray, np.void]):
     if isinstance(params, MutableMapping):
         return params
     try:
-        return {key: params[key] for key in params.dtype.names}
+        return {
+            key: params[key].item() if params[key].size == 1 else params[key]
+            for key in params.dtype.names
+        }
     except AttributeError:
         pass
     raise TypeError("params are not a Mapping")