Fix typos, create output_dir if it doesn't exit

scbirlab · Oct 5, 2024 · 79bb084 · 79bb084
1 parent 03e2ec2
commit 79bb084
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/carabiner/mpl/utils.py b/carabiner/mpl/utils.py
@@ -213,7 +213,7 @@ def scattergrid(
      return fig, axes
 
 def figsaver(
-     dir: str = ".",
+     output_dir: str = ".",
      prefix: Optional[str] = None,
      dpi: int = 300, 
      format: str = 'png', 
@@ -223,7 +223,7 @@ def figsaver(
 
      Parameters
      ----------
-     dir : str, optional
+     output_dir : str, optional
           Directory to save figures. Default: ".".
      prefix : str, optional
           Prefix for filenames. Default: no prefix.
@@ -241,6 +241,9 @@ def figsaver(
 
      """
 
+     if not os.path.exists(output_dir):
+          os.mkdir(output_dir)
+
      def _figsave(
           fig: figure.Figure, 
           name: str,