Merge pull request #192 from IBM/extend/symbolic

Testing tanh in symbolic expressions
IBM · Jul 26, 2024 · 23a953a · 23a953a
2 parents 0871d10 + df4ff63
commit 23a953a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/simulai/residuals/_pytorch_residuals.py b/simulai/residuals/_pytorch_residuals.py
@@ -72,7 +72,7 @@ def __init__(
         self.processing = processing
         self.periodic_bc_protected_key = "periodic"
 
-        self.protected_funcs = ["cos", "sin", "sqrt", "exp"]
+        self.protected_funcs = ["cos", "sin", "sqrt", "exp", "tanh"]
         self.protected_operators = ["L", "Div", "Identity", "Kronecker"]
 
         self.protected_funcs_subs = self._construct_protected_functions()

diff --git a/tests/residuals/test_symbolicoperator.py b/tests/residuals/test_symbolicoperator.py
@@ -171,7 +171,7 @@ def test_symbolic_buitin_functions(self):
         assert all([isinstance(item, torch.Tensor) for item in residual(input_data)])
 
     def test_symbolic_operator_ode(self):
-        for token in ["sin", "cos", "sqrt"]:
+        for token in ["sin", "cos", "sqrt", "tanh"]:
             f = f"D(u, t) - alpha*{token}(u)"
 
             input_labels = ["t"]