Import functools

pybamm/expression_tree/symbol.py

@@ -8,6 +8,8 @@
 import sympy
 from anytree.exporter import DotExporter
 from scipy.sparse import csr_matrix, issparse
+from functools import cache
+from functools import cached_property
 
 import pybamm
 from pybamm.expression_tree.printing.print_name import prettify_print_name

pybamm/models/base_model.py

@@ -7,6 +7,7 @@
 import copy
 import casadi
 import numpy as np
+from functools import cached_property
 
 import pybamm
 from pybamm.expression_tree.operations.latexify import Latexify

pybamm/models/full_battery_models/base_battery_model.py

@@ -4,6 +4,7 @@
 
 import pybamm
 import numbers
+from functools import cached_property
 
 
 class BatteryModelOptions(pybamm.FuzzyDict):

pybamm/models/full_battery_models/lithium_ion/electrode_soh.py

@@ -3,6 +3,7 @@
 #
 import pybamm
 import numpy as np
+from functools import cache
 
 
 class ElectrodeSOH(pybamm.BaseModel):

pybamm/simulation.py

@@ -7,6 +7,7 @@
 import copy
 import warnings
 import sys
+from functools import cache
 
 
 def is_notebook():

pybamm/solvers/solution.py

@@ -9,6 +9,7 @@
 import pybamm
 import pandas as pd
 from scipy.io import savemat
+from functools import cached_property
 
 
 class NumpyEncoder(json.JSONEncoder):