From ad9db9ab201c3a7c2c65bd555e8be3359eb120f1 Mon Sep 17 00:00:00 2001 From: dcaustin33 Date: Tue, 25 Jul 2023 11:21:30 -0700 Subject: [PATCH 1/4] Fixes ci tests in example notebooks --- dataquality/loggers/data_logger/base_data_logger.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dataquality/loggers/data_logger/base_data_logger.py b/dataquality/loggers/data_logger/base_data_logger.py index 4e847655b..e5c2c2cd8 100644 --- a/dataquality/loggers/data_logger/base_data_logger.py +++ b/dataquality/loggers/data_logger/base_data_logger.py @@ -262,7 +262,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 != 'semantic_segmentation': apply_umap_to_embs(location, last_epoch) else: print( From 84644c51891770d85902ce97657665d127171a8f Mon Sep 17 00:00:00 2001 From: dcaustin33 Date: Tue, 25 Jul 2023 11:30:47 -0700 Subject: [PATCH 2/4] Formatting --- dataquality/loggers/data_logger/base_data_logger.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dataquality/loggers/data_logger/base_data_logger.py b/dataquality/loggers/data_logger/base_data_logger.py index e5c2c2cd8..71f7f938b 100644 --- a/dataquality/loggers/data_logger/base_data_logger.py +++ b/dataquality/loggers/data_logger/base_data_logger.py @@ -262,7 +262,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() and config.task_type != 'semantic_segmentation': + if cuml_available() and config.task_type != "semantic_segmentation": apply_umap_to_embs(location, last_epoch) else: print( From e059257907c32677a86bda85279af7bd8966a667 Mon Sep 17 00:00:00 2001 From: dcaustin33 Date: Tue, 25 Jul 2023 11:31:41 -0700 Subject: [PATCH 3/4] Change --- dataquality/loggers/data_logger/base_data_logger.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dataquality/loggers/data_logger/base_data_logger.py b/dataquality/loggers/data_logger/base_data_logger.py index 71f7f938b..501908449 100644 --- a/dataquality/loggers/data_logger/base_data_logger.py +++ b/dataquality/loggers/data_logger/base_data_logger.py @@ -262,7 +262,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() and config.task_type != "semantic_segmentation": + if cuml_available() and config.task_type != TaskType.semantic_segmentation: apply_umap_to_embs(location, last_epoch) else: print( From 0fb7eced528dd7cd554c59c2993522f349f3e909 Mon Sep 17 00:00:00 2001 From: dcaustin33 Date: Tue, 25 Jul 2023 11:32:12 -0700 Subject: [PATCH 4/4] Fix --- dataquality/loggers/data_logger/base_data_logger.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dataquality/loggers/data_logger/base_data_logger.py b/dataquality/loggers/data_logger/base_data_logger.py index 501908449..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