Skip to content

Commit 8e9e8e8

Browse files
committed
fix diagnosis data rpc
1 parent 3b351e8 commit 8e9e8e8

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

dlrover/python/master/servicer.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -619,9 +619,7 @@ def _sync_checkpoint(
619619
def _report_worker_diagnosis_data(self, message: grpc.DiagnosisReportData):
620620
if self._diagnosis_manager:
621621
data_cls: Optional[DiagnosisData] = getattr(
622-
importlib.import_module(
623-
"dlrover.python.diagnosis.common.diagnosis_data"
624-
),
622+
self._diagnosis_data_module,
625623
message.data_cls,
626624
)
627625
if data_cls is None:

0 commit comments

Comments
 (0)