Fix logging

JustinShenk · Jun 15, 2023 · c3e9276 · c3e9276
1 parent ce06552
commit c3e9276
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/simages/__init__.py b/src/simages/__init__.py
@@ -6,7 +6,7 @@
 from .extractor import EmbeddingExtractor
 from .main import find_duplicates
 
-__version__ = "23.0.1"
+__version__ = "23.0.2"
 
 __title__ = "simages"
 __description__ = "Find similar images in a dataset"

diff --git a/src/simages/extractor.py b/src/simages/extractor.py
@@ -177,12 +177,12 @@ def is_valid(path):
         if not os.path.exists(embeddings_path):
             self.train()
             self.eval()
-            logging.info(f"saving embeddings to {embeddings_path}")
+            log.info(f"saving embeddings to {embeddings_path}")
             np.save(embeddings_path, self.embeddings)
         else:
-            logging.info("skipping training, loading embeddings from file")
+            log.info("skipping training, loading embeddings from file")
             self.embeddings = np.load(embeddings_path)
-            logging.info(f"Embeddings shape {self.embeddings.shape}")
+            log.info(f"Embeddings shape {self.embeddings.shape}")
 
 
     def _truncate_middle(self, string: str, n: int) -> str: