From b78fc3d012ed7fea5525fa19595fed52094145a9 Mon Sep 17 00:00:00 2001 From: dcaustin33 Date: Tue, 25 Jul 2023 12:23:04 -0700 Subject: [PATCH] Fixes ci tests in example notebooks (#725) Fixes an issue for those with cuml library installed where we tried to apply umap to embeddings even though semseg does not have any embeddings. --- dataquality/loggers/data_logger/base_data_logger.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dataquality/loggers/data_logger/base_data_logger.py b/dataquality/loggers/data_logger/base_data_logger.py index 4e847655b..899454870 100644 --- a/dataquality/loggers/data_logger/base_data_logger.py +++ b/dataquality/loggers/data_logger/base_data_logger.py @@ -21,6 +21,7 @@ from dataquality.schemas.dataframe import BaseLoggerDataFrames, DFVar from dataquality.schemas.ner import TaggingSchema from dataquality.schemas.split import Split +from dataquality.schemas.task_type import TaskType from dataquality.utils import tqdm from dataquality.utils.cloud import is_galileo_cloud from dataquality.utils.cuda import cuml_available @@ -262,7 +263,7 @@ def upload( proj_run = f"{config.current_project_id}/{config.current_run_id}" location = f"{self.LOG_FILE_DIR}/{proj_run}" - if cuml_available(): + if cuml_available() and config.task_type != TaskType.semantic_segmentation: apply_umap_to_embs(location, last_epoch) else: print(