提交 460326f4 编写于 作者: Y Yu Yang

Fix CI Test

上级 9f816352
...@@ -181,7 +181,7 @@ class OperatorFactory(object): ...@@ -181,7 +181,7 @@ class OperatorFactory(object):
self.op_methods = dict() self.op_methods = dict()
for op_proto in get_all_op_protos(): for op_proto in get_all_op_protos():
method = create_op_creation_method(op_proto) method = create_op_creation_method(op_proto)
self.op_methods[method['name']] = method self.op_methods[method.name] = method
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):
if 'type' in kwargs: if 'type' in kwargs:
......
...@@ -6,7 +6,6 @@ py_test(test_scope SRCS test_scope.py) ...@@ -6,7 +6,6 @@ py_test(test_scope SRCS test_scope.py)
py_test(test_tensor SRCS test_tensor.py) py_test(test_tensor SRCS test_tensor.py)
py_test(test_mul_op SRCS test_mul_op.py) py_test(test_mul_op SRCS test_mul_op.py)
py_test(test_network SRCS test_network.py)
py_test(test_mean_op SRCS test_mean_op.py) py_test(test_mean_op SRCS test_mean_op.py)
py_test(test_protobuf SRCS test_protobuf.py) py_test(test_protobuf SRCS test_protobuf.py)
......
...@@ -29,7 +29,7 @@ class OpTestMeta(type): ...@@ -29,7 +29,7 @@ class OpTestMeta(type):
for place in places: for place in places:
for in_name in Operator.get_op_input_names(self.type): for in_name in Operator.get_op_input_names(self.type):
if hasattr(self, "inputs") and in_name in self.inputs if hasattr(self, "inputs") and in_name in self.inputs:
kwargs[in_name] = in_name kwargs[in_name] = in_name
var = scope.new_var(in_name).get_tensor() var = scope.new_var(in_name).get_tensor()
arr = self.inputs[in_name] arr = self.inputs[in_name]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册