提交 8f1bb326 编写于 作者: L luopengting

set operation as None when it is empty or None

上级 72d3bb04
...@@ -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.
先完成此消息的编辑!
想要评论请 注册