Input checks added to set_source

RocketPy-Team · Nov 12, 2023 · 93dacbc · 93dacbc
1 parent f22cabc
commit 93dacbc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/rocketpy/mathutils/function.py b/rocketpy/mathutils/function.py
@@ -144,7 +144,7 @@ def set_source(self, source):
         self : Function
         """
         inputs, outputs, interpolation, extrapolation = self._check_user_input(
-            source, inputs, outputs, interpolation, extrapolation
+            source, self.__inputs__, self.__outputs__, self.__interpolation__, self.__extrapolation__
         )
         # If the source is a Function
         if isinstance(source, Function):
@@ -2676,6 +2676,8 @@ def _check_user_input(
         # check output type and dimensions
         if isinstance(outputs, str):
             outputs = [outputs]
+        if isinstance(inputs, str):
+            inputs = [inputs]
 
         elif len(outputs) > 1:
             raise ValueError(