Changed sindy optuna hyper parameter function to use types for featur…

…e library.
djpasseyjr · Jan 15, 2025 · bab0fdf · bab0fdf
1 parent b46a1ab
commit bab0fdf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/interfere/_methods/sindy.py b/interfere/_methods/sindy.py
@@ -12,7 +12,7 @@
 
 
 # Lists of hyper parameters for optimization.
-SINDY_LIB_LIST = [ps.PolynomialLibrary(), ps.FourierLibrary()]
+SINDY_LIB_LIST = [ps.PolynomialLibrary, ps.FourierLibrary]
 
 SINDY_DIFF_LIST = [
     ps.SINDyDerivative(kind='finite_difference', k=1),
@@ -44,7 +44,7 @@ def __init__(self,
         # Optionally accept types for feature library.
         if isinstance(feature_library, type):
             feature_library = feature_library()
-            
+
         # Differentiation method and feature library must be copied so that
         # their internal state doesn't carry over across different fits.
         if differentiation_method is not None: