Skip to content

Commit

Permalink
Fixes ci tests in example notebooks (#725)
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
dcaustin33 authored Jul 25, 2023
1 parent a2ffc95 commit b78fc3d
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion dataquality/loggers/data_logger/base_data_logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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(
Expand Down

0 comments on commit b78fc3d

Please sign in to comment.