Fix formatting

pysr/__main__.py

@@ -1,4 +1,4 @@
 from pysr._cli.main import pysr as _cli
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     _cli(prog_name="pysr")

pysr/_cli/main.py

@@ -8,29 +8,30 @@ def pysr(context):
     ctx = context
 
 
-@pysr.command("install",
-              help="Install Julia dependencies for PySR.")
-@click.option("-p",
-              "julia_project",
-              "--project",
-              default=None,
-              type=str,
-              help="Install in a specific Julia project (e.g., a local copy of SymbolicRegression.jl).",
-              metavar="PROJECT_DIRECTORY")
-@click.option("-q",
-              "--quiet",
-              is_flag=True,
-              default=False,
-              help="Disable logging.")
-@click.option("--precompile",
-              'precompile',
-              flag_value=True,
-              default=None,
-              help="Force precompilation of Julia libraries.")
-@click.option("--no-precompile",
-              'precompile',
-              flag_value=False,
-              default=None,
-              help="Disable precompilation.")
+@pysr.command("install", help="Install Julia dependencies for PySR.")
+@click.option(
+    "-p",
+    "julia_project",
+    "--project",
+    default=None,
+    type=str,
+    help="Install in a specific Julia project (e.g., a local copy of SymbolicRegression.jl).",
+    metavar="PROJECT_DIRECTORY",
+)
+@click.option("-q", "--quiet", is_flag=True, default=False, help="Disable logging.")
+@click.option(
+    "--precompile",
+    "precompile",
+    flag_value=True,
+    default=None,
+    help="Force precompilation of Julia libraries.",
+)
+@click.option(
+    "--no-precompile",
+    "precompile",
+    flag_value=False,
+    default=None,
+    help="Disable precompilation.",
+)
 def _install(julia_project, quiet, precompile):
     install(julia_project, quiet, precompile)

pysr/export_latex.py

@@ -96,7 +96,6 @@ def generate_single_table(
                         "$" + output_variable_name + " = " + latex_equation + "$"
                     )
                 else:
-
                     broken_latex_equation = " ".join(
                         [
                             r"\begin{minipage}{0.8\linewidth}",

pysr/test/test_cli.py

@@ -6,7 +6,11 @@ def run_command(command):
     """
     Retrieve output of a command string, decode and convert from CRLF to LF formatting
     """
-    return subprocess.run(command.split(" "), stdout=subprocess.PIPE).stdout.decode("utf-8").replace("\r\n", "\n")
+    return (
+        subprocess.run(command.split(" "), stdout=subprocess.PIPE)
+        .stdout.decode("utf-8")
+        .replace("\r\n", "\n")
+    )
 
 
 def make_command(command):
@@ -17,17 +21,18 @@ def make_command(command):
 
 
 class TestCli(unittest.TestCase):
-
     def test_help_on_all_commands(self):
         command_to_test = "python -m pysr --help"
-        expected_lines = ["Usage: pysr [OPTIONS] COMMAND [ARGS]...",
-                          "",
-                          "Options:",
-                          "  --help  Show this message and exit.",
-                          "",
-                          "Commands:",
-                          "  install  Install Julia dependencies for PySR.",
-                          ""]
+        expected_lines = [
+            "Usage: pysr [OPTIONS] COMMAND [ARGS]...",
+            "",
+            "Options:",
+            "  --help  Show this message and exit.",
+            "",
+            "Commands:",
+            "  install  Install Julia dependencies for PySR.",
+            "",
+        ]
 
         expected = make_command(expected_lines)
         actual = run_command(command_to_test)
@@ -35,19 +40,21 @@ class TestCli(unittest.TestCase):
 
     def test_help_on_install(self):
         command_to_test = "python -m pysr install --help"
-        expected_lines = ["Usage: pysr install [OPTIONS]",
-                          "",
-                          "  Install Julia dependencies for PySR.",
-                          "",
-                          "Options:",
-                          "  -p, --project PROJECT_DIRECTORY",
-                          "                                  Install in a specific Julia project (e.g., a",
-                          "                                  local copy of SymbolicRegression.jl).",
-                          "  -q, --quiet                     Disable logging.",
-                          "  --precompile                    Force precompilation of Julia libraries.",
-                          "  --no-precompile                 Disable precompilation.",
-                          "  --help                          Show this message and exit.",
-                          ""]
+        expected_lines = [
+            "Usage: pysr install [OPTIONS]",
+            "",
+            "  Install Julia dependencies for PySR.",
+            "",
+            "Options:",
+            "  -p, --project PROJECT_DIRECTORY",
+            "                                  Install in a specific Julia project (e.g., a",
+            "                                  local copy of SymbolicRegression.jl).",
+            "  -q, --quiet                     Disable logging.",
+            "  --precompile                    Force precompilation of Julia libraries.",
+            "  --no-precompile                 Disable precompilation.",
+            "  --help                          Show this message and exit.",
+            "",
+        ]
 
         expected = make_command(expected_lines)
         actual = run_command(command_to_test)