How can I use the TemplateExpressionSpec() function for a function that depends on a single variable?

[CleverCompiler-oss]

Hello! I am trying to determine the coefficient a in the sin(x) function, for the case when X is a vector using the following code:

import numpy as np
from pysr import PySRRegressor, TemplateExpressionSpec
X = np.random.randn(10, 1)
y = np.array(2.0001 * np.sin(X))
template = TemplateExpressionSpec(
    function_symbols=["a", "f"],
    combine="((; a, f), (x)) -> a() * sin(f(x))",
    #num_features = {"f": 1,},
)
model = PySRRegressor(
    expression_spec=template,
    binary_operators=["+"],
    maxsize=10,
) 
model.fit(X, y)

An error appears as a result of the execution.: “JuliaError: MethodError: no method matching sin(::Base.Iterators.Repeated{ValidVector{Vector{Float64}}}) The function sin exists, but no method is defined for this combination of argument types.”.
Can you tell me how to fix it? Thanks!

[MilesCranmer]

Ah, it's because you have to write (x,) instead of (x) - the former is a tuple (which is what we want - it uses tuple destructuring syntax), while the latter simplifies to just x.

[CleverCompiler-oss]

Everything is working! Thanks for the tip!