aeon: fix compat issue

bnediction · Jun 25, 2024 · 67f34b0 · 67f34b0
1 parent 3a1bb0e
commit 67f34b0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bonesis/aeon.py b/bonesis/aeon.py
@@ -201,6 +201,7 @@ def register_parameter(g):
                 else:
                     assert self.params[name] == args
                 return name
+            func = func.to_string(self.am) if not isinstance(func, str) else func
             f = self.ba.parse(RE_PARAMETER.sub(register_parameter, func))
             self._f[node] = f
             f = self._f[node]