pytorch/test/onnx/test_pytorch_onnx_onnxruntime.py
Lu Fang c1744a6c39 Add ONNX py3 CI cases (#21715)
Summary:
So far, we only have py2 ci for onnx. I think py3 support is important. And we have the plan to add onnxruntime backend tests, which only supports py3.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21715

Reviewed By: bddppq

Differential Revision: D15796885

Pulled By: houseroad

fbshipit-source-id: 8554dbb75d13c57b67ca054446a13a016983326c
2019-06-14 10:20:14 -07:00

19 lines
394 B
Python

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
import unittest
import sys
import onnxruntime # noqa
class TestONNXRuntime(unittest.TestCase):
def test_onnxruntime_installed(self):
self.assertTrue('onnxruntime' in sys.modules)
if __name__ == '__main__':
unittest.main()