Fixed bug: report succes of output to logger properly

beeldengeluid · Nov 24, 2023 · 9241b48 · 9241b48
1 parent 9f56d9d
commit 9241b48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/feature_extraction.py b/feature_extraction.py
@@ -116,7 +116,7 @@ def _save_features_to_file(features: torch.Tensor, output_file_path: str) -> boo
         if not os.path.isdir(parent_dir):
             logger.info("Parent dir, did not exist, creating it now")
             os.makedirs(parent_dir)
-        np.save(output_file_path, np.array(features))
+        return True
     except Exception:
         logger.exception("Failed to save features to file")
-    return False
+        return False