BUG: savetxt accepts lambdas

MateusStano · MateusStano · commit 48378d55b46e · 2024-10-01T16:30:23.000+02:00
diff --git a/rocketpy/mathutils/function.py b/rocketpy/mathutils/function.py
@@ -3171,7 +3171,7 @@ def savetxt(
                 )
             # Generate the data points using the callable
             x = np.linspace(lower, upper, samples)
-            data_points = np.column_stack((x, self.source(x)))
+            data_points = np.column_stack((x, self.get_value_opt(x)))
         else:
             # If the source is already an array, use it as is
             data_points = self.source

Original file line number	Diff line number	Diff line change
`@@ -3171,7 +3171,7 @@ def savetxt(`
`3171`	`3171`	`)`
`3172`	`3172`	`# Generate the data points using the callable`
`3173`	`3173`	`x = np.linspace(lower, upper, samples)`
`3174`		`- data_points = np.column_stack((x, self.source(x)))`
	`3174`	`+ data_points = np.column_stack((x, self.get_value_opt(x)))`
`3175`	`3175`	`else:`
`3176`	`3176`	`# If the source is already an array, use it as is`
`3177`	`3177`	`data_points = self.source`