Update utils

yann_utils/phd.py

@@ -2,14 +2,14 @@
 import subprocess
 import os
 
-def csv_to_latex_png(df, png_filename):
+def csv_to_latex_png(df, png_filename, caption):
     # 1. Convert the df to LaTeX table format
     latex_code = r"""
     \documentclass{standalone}
     \usepackage{booktabs}
     \begin{document}
     """
-    latex_code += df.to_latex(index=False, escape=False)
+    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:

yann_utils/utils.py

@@ -107,6 +107,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] = []
 
@@ -130,3 +131,14 @@ def load(self, name: str) -> Any:
 
   def __repr__(self):
     return "\n".join(self.locs)
+
+def set_seed(seed):
+  import random
+  import numpy as np
+  import torch
+
+  random.seed(seed)
+  np.random.seed(seed)
+  torch.manual_seed(seed)
+  torch.cuda.manual_seed_all(seed)
+  # if torch is installed