fixed erroneous indent in save function

sparks-baird · Sep 9, 2024 · bd5c58b · bd5c58b
1 parent b9835dc
commit bd5c58b
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}")