onnxruntime_extensions

Signed-off-by: xavier dupré <[email protected]>

Author: sdpython

ci_build/azure_pipelines/templates/setup.yml

@@ -27,7 +27,7 @@ steps:
 
     if [[ $CI_PLATFORM == "windows" ]] ;
     then
-      pip install -i https://test.pypi.org/simple/ onnxruntime-customops==0.0.1
+      pip install onnxruntime-extensions
       if [[ $CI_TF_VERSION == 2.* ]] ;
       then
         pip install tensorflow-text

tests/backend_test_base.py

@@ -80,7 +80,7 @@ def run_onnxruntime(self, model_path, inputs, output_names, use_custom_ops=False
                 providers = ['CUDAExecutionProvider']
         opt = rt.SessionOptions()
         if use_custom_ops:
-            from onnxruntime_customops import get_library_path
+            from onnxruntime_extensions import get_library_path
             opt.register_custom_ops_library(get_library_path())
         # in case of issues with the runtime, one can enable more logging
         # opt.log_severity_level = 0

tests/common.py

@@ -242,7 +242,7 @@ def requires_custom_ops(message=""):
     """ Skip until custom ops framework is on PyPI. """
     reason = _append_message("test needs custom ops framework", message)
     try:
-        import onnxruntime_customops  #pylint: disable=import-outside-toplevel,unused-import
+        import onnxruntime_extensions  #pylint: disable=import-outside-toplevel,unused-import
         can_import = True
     except ModuleNotFoundError:
         can_import = False