1.9 add plotting convenience function

HastingsGreer · Jan 5, 2024 · 121177f · 121177f
1 parent e8706e3
commit 121177f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = footsteps
-version = 0.1.8
+version = 0.1.9
 author = Hastings Greer
 author_email = [email protected]
 description = a minimal experiment logging package

diff --git a/src/footsteps/__init__.py b/src/footsteps/__init__.py
@@ -135,3 +135,11 @@ def __getattr__(name):
             initialize()
         return output_dir_impl
     raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
+
+num_plots = 0
+def plot(name=""):
+    global num_plots
+    import matplotlib.pyplot as plt
+    plt.savefig(__getattr__("output_dir") + f"{num_plots:03d}_{name}.png")
+    plt.clf()
+    num_plots += 1