Skip to content

Commit

Permalink
Add ONNX docs to Python API section (mlflow#1388)
Browse files Browse the repository at this point in the history
  • Loading branch information
dbczumar authored and aarondav committed Jun 3, 2019
1 parent b4a7598 commit d9fa1c9
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 1 deletion.
7 changes: 7 additions & 0 deletions docs/source/python_api/mlflow.onnx.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
mlflow.onnx
==================

.. automodule:: mlflow.onnx
:members:
:undoc-members:
:show-inheritance:
12 changes: 11 additions & 1 deletion mlflow/onnx.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
"""
The ``mlflow.onnx`` module provides APIs for logging and loading ONNX models in the MLflow Model
format. This module exports MLflow Models with the following flavors:
ONNX (native) format
This is the main flavor that can be loaded back as an ONNX model object.
:py:mod:`mlflow.pyfunc`
Produced for use by generic pyfunc-based deployment tools and batch inference.
"""

from __future__ import absolute_import

import os
Expand All @@ -23,7 +33,7 @@
def get_default_conda_env():
"""
:return: The default Conda environment for MLflow Models produced by calls to
:func:`save_model()` and :func:`log_model()`.
:func:`save_model()` and :func:`log_model()`.
"""
import onnx
import onnxruntime
Expand Down

0 comments on commit d9fa1c9

Please sign in to comment.