From 076e7a9fd8aacdfb08dd86cb9ef3e16a7a6b6589 Mon Sep 17 00:00:00 2001 From: jiangshuqiang Date: Mon, 27 Sep 2021 20:36:04 +0800 Subject: [PATCH] add warning for none operation --- mindspore/train/callback/_dataset_graph.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mindspore/train/callback/_dataset_graph.py b/mindspore/train/callback/_dataset_graph.py index 224d53b27ea..4a9914a2169 100644 --- a/mindspore/train/callback/_dataset_graph.py +++ b/mindspore/train/callback/_dataset_graph.py @@ -100,6 +100,10 @@ class DatasetGraph: operation (dict): Enhancement operation. message (Operation): Enhancement operation proto message. """ + if operation is None: + logger.warning("Summary can not collect the operation for dataset graph as the operation is None, " + "please check the dataset.") + return for key, value in operation.items(): if isinstance(value, (list, tuple)): if all(isinstance(ele, int) for ele in value):