Coverage for mlprodict/onnxrt/ops_cpu/op_sum.py: 100%

1# -*- encoding: utf-8 -*- 

2# pylint: disable=E0203,E1101,C0111 

3""" 

4@file 

5@brief Runtime operator. 

6""" 

7from ._op import OpRun 

8 

9 

10class Sum(OpRun): 

11 

12 def __init__(self, onnx_node, desc=None, **options): 

13 OpRun.__init__(self, onnx_node, desc=desc, **options) 

14 

15 def _run(self, *args, attributes=None, verbose=0, fLOG=None): # pylint: disable=W0221 

16 return (sum(args), ) 

17 

18 def to_python(self, inputs): 

19 return None, f"return sum([{', '.join(inputs)}])"