From a5c2b1f75335287bf130b3235f4348d45aef69c3 Mon Sep 17 00:00:00 2001 From: Mrtutu Date: Tue, 6 Feb 2024 10:40:45 +0800 Subject: [PATCH] cluster bug --- .../advisor_backend/cluster_advice/kernel_cluster_advice.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/profiler/advisor/advisor_backend/cluster_advice/kernel_cluster_advice.py b/profiler/advisor/advisor_backend/cluster_advice/kernel_cluster_advice.py index 644d46714..e2ca914a7 100644 --- a/profiler/advisor/advisor_backend/cluster_advice/kernel_cluster_advice.py +++ b/profiler/advisor/advisor_backend/cluster_advice/kernel_cluster_advice.py @@ -2,6 +2,7 @@ import os import pandas as pd from common_func.path_manager import PathManager from common_func.constant import Constant +from common_func_advisor.constant import Constant as AdvisorConstant from cluster_advice.cluster_advice_base import ClusterAdviceBase from cluster_data_preprocess.pytorch_data_preprocessor import PytorchDataPreprocessor @@ -55,7 +56,7 @@ class KernelClusterAdvice(ClusterAdviceBase): def get_prof_dirs(self, collection_path): prof_dirs = [] for prof_dir in os.listdir(collection_path): - if prof_dir.endswith(Constant.PT_PROF_SUFFIX): + if prof_dir.endswith(AdvisorConstant.PT_PROF_SUFFIX): prof_dirs.append(os.path.join(collection_path, prof_dir)) return prof_dirs \ No newline at end of file -- Gitee