pysr/__init__.py

from . import sklearn_monkeypatch
from .version import __version__
from .sr import (
    pysr,
    PySRRegressor,
    best,
    best_tex,
    best_callable,
    best_row,
)
from .julia_helpers import install
from .feynman_problems import Problem, FeynmanProblem
from .export_jax import sympy2jax
from .export_torch import sympy2torch

__all__ = [
    "sklearn_monkeypatch",
    "__version__",
    "pysr",
    "PySRRegressor",
    "best",
    "best_tex",
    "best_callable",
    "best_row",
    "install",
    "Problem",
    "FeynmanProblem",
    "sympy2jax",
    "sympy2torch",
]