diff --git a/examples/scripts/functional_parameters.py b/examples/scripts/functional_parameters.py index 37f82102..cc2d9e4d 100644 --- a/examples/scripts/functional_parameters.py +++ b/examples/scripts/functional_parameters.py @@ -1,7 +1,7 @@ import numpy as np -import pybop import pybamm +import pybop # This example demonstrates how to use a pybamm.FunctionalParameter to # optimise functional parameters using PyBOP. @@ -14,6 +14,7 @@ parameter_set = pybop.ParameterSet.pybamm("Chen2020") model = pybop.lithium_ion.SPM(parameter_set=parameter_set) + # Define a new function using new parameters def positive_electrode_exchange_current_density(c_e, c_s_surf, c_s_max, T): # New parameters @@ -39,7 +40,9 @@ def positive_electrode_exchange_current_density(c_e, c_s_surf, c_s_max, T): }, check_already_exists=False, ) -parameter_set["Positive electrode exchange-current density [A.m-2]"] = positive_electrode_exchange_current_density +parameter_set["Positive electrode exchange-current density [A.m-2]"] = ( + positive_electrode_exchange_current_density +) # Fitting parameters parameters = pybop.Parameters( @@ -71,7 +74,7 @@ def positive_electrode_exchange_current_density(c_e, c_s_surf, c_s_max, T): # Generate problem, cost function, and optimisation class problem = pybop.FittingProblem(model, parameters, dataset) cost = pybop.RootMeanSquaredError(problem) -optim = pybop.SciPyMinimize(cost,max_iterations=125) +optim = pybop.SciPyMinimize(cost, max_iterations=125) # Run optimisation x, final_cost = optim.run()