From 8f1bb3269f70081e4ff8851bdb25d0c33fa73940 Mon Sep 17 00:00:00 2001 From: luopengting Date: Tue, 28 Apr 2020 17:44:51 +0800 Subject: [PATCH] set operation as None when it is empty or None --- mindinsight/lineagemgr/summary/_summary_adapter.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/mindinsight/lineagemgr/summary/_summary_adapter.py b/mindinsight/lineagemgr/summary/_summary_adapter.py index f5b02605..c5adb1ef 100644 --- a/mindinsight/lineagemgr/summary/_summary_adapter.py +++ b/mindinsight/lineagemgr/summary/_summary_adapter.py @@ -70,13 +70,13 @@ def _package_current_dataset(operation, message): message (Operation): Operation proto message. """ for key, value in operation.items(): - if key == "operations": + if value and key == "operations": for operator in value: _package_enhancement_operation( operator, message.operations.add() ) - elif key == "sampler": + elif value and key == "sampler": _package_enhancement_operation( value, message.sampler @@ -93,7 +93,6 @@ def _package_enhancement_operation(operation, message): operation (dict): Enhancement operation. message (Operation): Enhancement operation proto message. """ - for key, value in operation.items(): if isinstance(value, list): if all(isinstance(ele, int) for ele in value):