Add pre-commit hooks for unused imports

.pre-commit-config.yaml

@@ -18,3 +18,8 @@ repos:
     rev: 0.6.1
     hooks:
       - id: nbstripout
+  # Unused imports:
+  - repo: https://github.com/hadialqattan/pycln
+    rev: "v2.2.2"
+    hooks:
+      - id: pycln

benchmarks/print_best_model.py

@@ -1,5 +1,4 @@
 """Print the best model parameters and loss"""
-import sys
 import numpy as np
 import pickle as pkl
 import hyperopt

docs/gen_param_docs.py

@@ -1,5 +1,4 @@
 # Load YAML file param_groupings.yml:
-from pathlib import Path
 from yaml import safe_load
 import sys
 

pysr/__init__.py

@@ -12,3 +12,19 @@ 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",
+]

pysr/export_jax.py

@@ -1,7 +1,4 @@
-import functools as ft
 import sympy
-import string
-import random
 
 # Special since need to reduce arguments.
 MUL = 0

pysr/export_latex.py

@@ -3,7 +3,6 @@ import sympy
 from sympy.printing.latex import LatexPrinter
 import pandas as pd
 from typing import List
-import warnings
 
 
 class PreciseLatexPrinter(LatexPrinter):

pysr/test/__init__.py

@@ -3,3 +3,5 @@ from .test_env import runtests as runtests_env
 from .test_jax import runtests as runtests_jax
 from .test_torch import runtests as runtests_torch
 from .test_cli import runtests as runtests_cli
+
+__all__ = ["runtests", "runtests_env", "runtests_jax", "runtests_torch", "runtests_cli"]