fix scale and shift dispatch methods for single string input keys

Author: LarsKue

bayesflow/adapters/adapter.py

@@ -558,12 +558,18 @@ def rename(self, from_key: str, to_key: str):
     def scale(self, keys: str | Sequence[str], by: float | np.ndarray):
         from .transforms import Scale
 
+        if isinstance(keys, str):
+            keys = [keys]
+
         self.transforms.append(MapTransform({key: Scale(scale=by) for key in keys}))
         return self
 
     def shift(self, keys: str | Sequence[str], by: float | np.ndarray):
         from .transforms import Shift
 
+        if isinstance(keys, str):
+            keys = [keys]
+
         self.transforms.append(MapTransform({key: Shift(shift=by) for key in keys}))
         return self