fix: s2s embedding bug (#858)

rungalileo · Jun 3, 2024 · 15b7a8d · 15b7a8d
1 parent fbe15bc
commit 15b7a8d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/dataquality/__init__.py b/dataquality/__init__.py
@@ -30,7 +30,7 @@
         dataquality.get_insights()
 """
 
-__version__ = "2.1.1"
+__version__ = "2.1.2"
 
 import sys
 from typing import Any, List, Optional

diff --git a/dataquality/loggers/data_logger/base_data_logger.py b/dataquality/loggers/data_logger/base_data_logger.py
@@ -265,7 +265,8 @@ def upload(
         location = f"{self.LOG_FILE_DIR}/{proj_run}"
 
         if cuml_available():
-            if self.support_embs:
+            # Only run UMAP if we have embeddings and we won't remove them later
+            if self.support_embs and not self.logger_config.remove_embs:
                 print("Applying UMAP to embeddings")
                 apply_umap_to_embs(location, last_epoch)
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "dataquality"
-version = "2.1.1"
+version = "2.1.2"
 description = ""
 authors = ["Galileo Technologies, Inc. <[email protected]>"]
 readme = "README.md"