Merge pull request #698 from RocketPy-Team/bug/savetxt-lambda-functions

MateusStano · web-flow · commit ffcb1ef975ef · 2024-10-01T19:02:17.000+02:00
BUG: `savetxt` Not Accepting lambda Functions
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`