From d5b45f5d9fcb2742242ce5df5c93cdb217ca0ad9 Mon Sep 17 00:00:00 2001 From: liuht-0807 Date: Mon, 4 Dec 2023 20:44:29 +0800 Subject: [PATCH] [FIX] delete loguru --- .../market/heterogeneous/organizer/hetero_map/__init__.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/learnware/market/heterogeneous/organizer/hetero_map/__init__.py b/learnware/market/heterogeneous/organizer/hetero_map/__init__.py index 6735879..4ebb0a8 100644 --- a/learnware/market/heterogeneous/organizer/hetero_map/__init__.py +++ b/learnware/market/heterogeneous/organizer/hetero_map/__init__.py @@ -9,9 +9,7 @@ from torch import Tensor, nn from .....utils import allocate_cuda_idx, choose_device from .....specification import HeteroMapTableSpecification, RKMETableSpecification from .feature_extractor import CLSToken, FeatureProcessor, FeatureTokenizer -from .trainer import Trainer, TransTabCollatorForCL - -from loguru import logger +from .trainer import TransTabCollatorForCL class HeteroMap(nn.Module):