diff --git a/examples/tuning/cta_singlecellnet/main.py b/examples/tuning/cta_singlecellnet/main.py index 9ac25de7..892c6633 100644 --- a/examples/tuning/cta_singlecellnet/main.py +++ b/examples/tuning/cta_singlecellnet/main.py @@ -6,8 +6,8 @@ from typing import get_args import numpy as np - import wandb + from dance import logger from dance.datasets.singlemodality import CellTypeAnnotationDataset from dance.modules.single_modality.cell_type_annotation.singlecellnet import SingleCellNet