Feat: Allow access of params inside function prescription

NiallOswald · Feb 25, 2025 · 534fc5b · 534fc5b
1 parent af53b4d
commit 534fc5b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/photosurfactant/scripts/plot_first_order.py b/photosurfactant/scripts/plot_first_order.py
@@ -326,13 +326,13 @@ def plot_first_order():  # noqa: D103
     first_order_parser(parser)
     args = parser.parse_args()
 
-    root_index = args.root_index
-    func = eval("lambda x, L: " + args.func)
-    problem = args.problem
-
     params = Parameters.from_dict(vars(args))
     plot_params = PlottingParameters.from_dict(vars(args))
 
+    root_index = args.root_index
+    func = eval("lambda x: " + args.func)
+    problem = args.problem
+
     # Calculate Fourier series coefficients
     if args.mollify:
         wavenumbers, func_coeffs = convolution_coeff(