Clean up

pyproject.toml

@@ -17,3 +17,6 @@ pytest = "^7.4.0"
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.ruff]
+indent-width=2

tests/test_chunking.py

@@ -3,8 +3,7 @@
 import numpy as np
 import pytest
 
-from yann_utils.chunking import (chunk_obj, get_chunks, persist_chunks,
-                                 reconstruct_obj)
+from yann_utils.chunking import chunk_obj, get_chunks, persist_chunks, reconstruct_obj
 
 
 @pytest.fixture

yann_utils/phd.py

@@ -2,28 +2,29 @@
 import subprocess
 import os
 
+
 def csv_to_latex_png(df, png_filename, caption):
-    # 1. Convert the df to LaTeX table format
-    latex_code = r"""
+  # 1. Convert the df to LaTeX table format
+  latex_code = r"""
     \documentclass{standalone}
     \usepackage{booktabs}
     \begin{document}
     """
-    latex_code += df.to_latex(index=True, escape=False, caption=caption)
-    latex_code += r"\end{document}"
+  latex_code += df.to_latex(index=True, escape=False, caption=caption)
+  latex_code += r"\end{document}"
 
-    with open("temp_table.tex", "w") as latex_file:
-        latex_file.write(latex_code)
+  with open("temp_table.tex", "w") as latex_file:
+    latex_file.write(latex_code)
 
-    # 2. Compile the LaTeX to produce a PDF
-    subprocess.call(["pdflatex", "-interaction=nonstopmode", "temp_table.tex"])
+  # 2. Compile the LaTeX to produce a PDF
+  subprocess.call(["pdflatex", "-interaction=nonstopmode", "temp_table.tex"])
 
-    # 3. Convert the PDF to PNG
-    subprocess.call(["pdftoppm", "-png", "temp_table.pdf", "temp_table"])
+  # 3. Convert the PDF to PNG
+  subprocess.call(["pdftoppm", "-png", "temp_table.pdf", "temp_table"])
 
-    # Rename to the desired png filename
-    os.rename("temp_table-1.png", png_filename)
+  # Rename to the desired png filename
+  os.rename("temp_table-1.png", png_filename)
 
-    # Clean up intermediate files
-    for ext in [".tex", ".aux", ".log", ".pdf"]:
-        os.remove("temp_table" + ext)
+  # Clean up intermediate files
+  for ext in [".tex", ".aux", ".log", ".pdf"]:
+    os.remove("temp_table" + ext)

yann_utils/utils.py

@@ -8,6 +8,7 @@
 from typing import Dict, Generator, Tuple
 import pickle
 from tqdm import tqdm
+import os
 
 
 A = TypeVar("A")
@@ -107,6 +108,7 @@ def enumerate_flipped(xs: List["E"]) -> Generator[Tuple["E", int], None, None]:
 
 class Cache:
   """Persistent and automated caching of objects."""
+
   import os
 
   locs: List[str] = []
@@ -132,6 +134,7 @@ def load(self, name: str) -> Any:
   def __repr__(self):
     return "\n".join(self.locs)
 
+
 def set_seed(seed):
   import random
   import numpy as np
@@ -141,4 +144,3 @@ def set_seed(seed):
   np.random.seed(seed)
   torch.manual_seed(seed)
   torch.cuda.manual_seed_all(seed)
-  # if torch is installed