Coverage for mlprodict/npy/numpy_onnx_pyrt_skl.py: 100%

1""" 

2@file 

3@brief :epkg:`numpy` functions implemented with :epkg:`onnx` 

4and compiled with this python runtime. 

5 

6.. versionadded:: 0.6 

7""" 

8import numpy 

9from .onnx_numpy_annotation import NDArrayType 

10from .numpy_onnx_impl_skl import ( 

11 logistic_regression as nx_logistic_regression) 

12from .onnx_numpy_wrapper import onnxnumpy_np 

13 

14 

15@onnxnumpy_np(signature=NDArrayType(("T:all", ), dtypes_out=((numpy.int64,), "T"))) 

16def logistic_regression(x, *, model=None): 

17 "logistic_regression" 

18 return nx_logistic_regression(x, model=model)