Add pre-commit config

.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]

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",

fonts/OFL.txt

@@ -18,7 +18,7 @@ with others.
 
 The OFL allows the licensed fonts to be used, studied, modified and
 redistributed freely as long as they are not sold by themselves. The
-fonts, including any derivative works, can be bundled, embedded, 
+fonts, including any derivative works, can be bundled, embedded,
 redistributed and/or sold with any software provided that any reserved
 names are not used by derivative works. The fonts and derivatives,
 however, cannot be released under any other type of license. The

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