Add pre-commit config

gui/.pre-commit-config.yaml

@@ -0,0 +1,33 @@
+repos:
+  # General linting
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.5.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-yaml
+      - id: check-added-large-files
+  # General formatting
+  - repo: https://github.com/psf/black
+    rev: 23.12.1
+    hooks:
+      - id: black
+      - id: black-jupyter
+        exclude: pysr/test/test_nb.ipynb
+  # Stripping notebooks
+  - repo: https://github.com/kynan/nbstripout
+    rev: 0.6.1
+    hooks:
+      - id: nbstripout
+        exclude: pysr/test/test_nb.ipynb
+  # Unused imports
+  - repo: https://github.com/hadialqattan/pycln
+    rev: "v2.4.0"
+    hooks:
+      - id: pycln
+  # Sorted imports
+  - repo: https://github.com/PyCQA/isort
+    rev: "5.13.2"
+    hooks:
+      - id: isort
+        additional_dependencies: [toml]

gui/app.py

@@ -1,14 +1,14 @@
+import multiprocessing as mp
+import os
+import time
+
 import gradio as gr
 import numpy as np
-import os
 import pandas as pd
-import time
-import multiprocessing as mp
 from matplotlib import pyplot as plt
 
 plt.ioff()
 import tempfile
-from typing import Optional, Union
 from pathlib import Path
 
 empty_df = pd.DataFrame(
@@ -25,7 +25,7 @@ test_equations = ["sin(2*x)/x + 0.1*x"]
 def generate_data(s: str, num_points: int, noise_level: float, data_seed: int):
     rstate = np.random.RandomState(data_seed)
     x = rstate.uniform(-10, 10, num_points)
-    for (k, v) in {
+    for k, v in {
         "sin": "np.sin",
         "cos": "np.cos",
         "exp": "np.exp",

gui/gen_example_data.py

@@ -1,5 +1,5 @@
-import pandas as pd
 import numpy as np
+import pandas as pd
 
 rand_between = lambda a, b, size: np.random.rand(*size) * (b - a) + a