提交 c79adb4d 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!85 Set operation as 'None' when it is empty or None

Merge pull request !85 from luopengting/lineage_record
...@@ -70,13 +70,13 @@ def _package_current_dataset(operation, message): ...@@ -70,13 +70,13 @@ def _package_current_dataset(operation, message):
message (Operation): Operation proto message. message (Operation): Operation proto message.
""" """
for key, value in operation.items(): for key, value in operation.items():
if key == "operations": if value and key == "operations":
for operator in value: for operator in value:
_package_enhancement_operation( _package_enhancement_operation(
operator, operator,
message.operations.add() message.operations.add()
) )
elif key == "sampler": elif value and key == "sampler":
_package_enhancement_operation( _package_enhancement_operation(
value, value,
message.sampler message.sampler
...@@ -93,7 +93,6 @@ def _package_enhancement_operation(operation, message): ...@@ -93,7 +93,6 @@ def _package_enhancement_operation(operation, message):
operation (dict): Enhancement operation. operation (dict): Enhancement operation.
message (Operation): Enhancement operation proto message. message (Operation): Enhancement operation proto message.
""" """
for key, value in operation.items(): for key, value in operation.items():
if isinstance(value, list): if isinstance(value, list):
if all(isinstance(ele, int) for ele in value): if all(isinstance(ele, int) for ele in value):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册