Merge pull request #76 from kyledmiller/save_fix

fixed erroneous indent in save function
sparks-baird · Sep 9, 2024 · bd0f7b6 · bd0f7b6
2 parents b9835dc + bd5c58b
commit bd0f7b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crabnet/crabnet_.py b/crabnet/crabnet_.py
@@ -1094,8 +1094,8 @@ def save_network(self, model_name: str = None):
         """
         if model_name is None:
             model_name = self.model_name
-            os.makedirs(join("models", "trained_models"), exist_ok=True)
-            path = join("models", "trained_models", f"{model_name}.pth")
+        os.makedirs(join("models", "trained_models"), exist_ok=True)
+        path = join("models", "trained_models", f"{model_name}.pth")
         if self.verbose:
             print(f"Saving network ({model_name}) to {path}")