AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_reduce_sum_square_keepdims_example' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_reduce_sum_square_keepdims_example.py'))
----
import numpy
from onnx import TensorProto
from onnx.helper import make_tensor
from onnxscript import script, external_tensor
from onnxscript.values import Opset
from onnxscript.onnx_types import FLOAT, INT64
from onnxscript.onnx_opset import opset18

@script()
def bck_test_reduce_sum_square_keepdims_example(data: FLOAT[3,2,2], axes: INT64[1]) -> (FLOAT[3,1,2]):
    reduced = opset18.ReduceSumSquare(data, axes, keepdims=1)
    return reduced
onnxscript\backend\onnx_export_test.py:116: in extract_functions
    mod = importlib.import_module(import_name)
C:\hostedtoolcache\windows\Python\3.8.10\x64\lib\importlib\__init__.py:127: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
E   ModuleNotFoundError: No module named 'onnxscript.tests.onnx_backend_test_code.test_reduce_sum_square_keepdims_example'

The above exception was the direct cause of the following exception:
.nox\test\lib\site-packages\parameterized\parameterized.py:620: in standalone_func
    return func(*(a + p.args), **p.kwargs, **kw)
onnxscript\backend\onnx_export_test.py:235: in test_export2python_produces_correct_onnx_script_model
    functions = extract_functions(backend_test.name, code, self.test_folder)
onnxscript\backend\onnx_export_test.py:118: in extract_functions
    raise AssertionError(
E   AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_reduce_sum_square_keepdims_example' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_reduce_sum_square_keepdims_example.py'))
E   ----
E   import numpy
E   from onnx import TensorProto
E   from onnx.helper import make_tensor
E   from onnxscript import script, external_tensor
E   from onnxscript.values import Opset
E   from onnxscript.onnx_types import FLOAT, INT64
E   from onnxscript.onnx_opset import opset18
E   
E   @script()
E   def bck_test_reduce_sum_square_keepdims_example(data: FLOAT[3,2,2], axes: INT64[1]) -> (FLOAT[3,1,2]):
E       reduced = opset18.ReduceSumSquare(data, axes, keepdims=1)
E       return reduced

AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_argmin_keepdims_random_select_last_index' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_argmin_keepdims_random_select_last_index.py'))
----
import numpy
from onnx import TensorProto
from onnx.helper import make_tensor
from onnxscript import script, external_tensor
from onnxscript.values import Opset
from onnxscript.onnx_types import FLOAT, INT64
from onnxscript.onnx_opset import opset13

@script()
def bck_test_argmin_keepdims_random_select_last_index(data: FLOAT[2,3,4]) -> (INT64[2,1,4]):
    result = opset13.ArgMin(data, axis=1, keepdims=1, select_last_index=1)
    return result
onnxscript\backend\onnx_export_test.py:116: in extract_functions
    mod = importlib.import_module(import_name)
C:\hostedtoolcache\windows\Python\3.8.10\x64\lib\importlib\__init__.py:127: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
E   ModuleNotFoundError: No module named 'onnxscript.tests.onnx_backend_test_code.test_argmin_keepdims_random_select_last_index'

The above exception was the direct cause of the following exception:
.nox\test\lib\site-packages\parameterized\parameterized.py:620: in standalone_func
    return func(*(a + p.args), **p.kwargs, **kw)
onnxscript\backend\onnx_export_test.py:235: in test_export2python_produces_correct_onnx_script_model
    functions = extract_functions(backend_test.name, code, self.test_folder)
onnxscript\backend\onnx_export_test.py:118: in extract_functions
    raise AssertionError(
E   AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_argmin_keepdims_random_select_last_index' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_argmin_keepdims_random_select_last_index.py'))
E   ----
E   import numpy
E   from onnx import TensorProto
E   from onnx.helper import make_tensor
E   from onnxscript import script, external_tensor
E   from onnxscript.values import Opset
E   from onnxscript.onnx_types import FLOAT, INT64
E   from onnxscript.onnx_opset import opset13
E   
E   @script()
E   def bck_test_argmin_keepdims_random_select_last_index(data: FLOAT[2,3,4]) -> (INT64[2,1,4]):
E       result = opset13.ArgMin(data, axis=1, keepdims=1, select_last_index=1)
E       return result

AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_greater_equal_bcast_expanded' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_greater_equal_bcast_expanded.py'))
----
import numpy
from onnx import TensorProto
from onnx.helper import make_tensor
from onnxscript import script, external_tensor
from onnxscript.values import Opset
from onnxscript.onnx_types import BOOL, FLOAT
from onnxscript.onnx_opset import opset16

@script()
def bck_test_greater_equal_bcast_expanded(x: FLOAT[3,4,5], y: FLOAT[5]) -> (BOOL[3,4,5]):
    GreaterOrEqual_test_greater_equal_bcast_expanded_function_O1 = opset16.Greater(x, y)
    GreaterOrEqual_test_greater_equal_bcast_expanded_function_O2 = opset16.Equal(x, y)
    greater_equal = opset16.Or(GreaterOrEqual_test_greater_equal_bcast_expanded_function_O1, GreaterOrEqual_test_greater_equal_bcast_expanded_function_O2)
    return greater_equal
onnxscript\backend\onnx_export_test.py:116: in extract_functions
    mod = importlib.import_module(import_name)
C:\hostedtoolcache\windows\Python\3.8.10\x64\lib\importlib\__init__.py:127: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
E   ModuleNotFoundError: No module named 'onnxscript.tests.onnx_backend_test_code.test_greater_equal_bcast_expanded'

The above exception was the direct cause of the following exception:
.nox\test\lib\site-packages\parameterized\parameterized.py:620: in standalone_func
    return func(*(a + p.args), **p.kwargs, **kw)
onnxscript\backend\onnx_export_test.py:235: in test_export2python_produces_correct_onnx_script_model
    functions = extract_functions(backend_test.name, code, self.test_folder)
onnxscript\backend\onnx_export_test.py:118: in extract_functions
    raise AssertionError(
E   AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_greater_equal_bcast_expanded' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_greater_equal_bcast_expanded.py'))
E   ----
E   import numpy
E   from onnx import TensorProto
E   from onnx.helper import make_tensor
E   from onnxscript import script, external_tensor
E   from onnxscript.values import Opset
E   from onnxscript.onnx_types import BOOL, FLOAT
E   from onnxscript.onnx_opset import opset16
E   
E   @script()
E   def bck_test_greater_equal_bcast_expanded(x: FLOAT[3,4,5], y: FLOAT[5]) -> (BOOL[3,4,5]):
E       GreaterOrEqual_test_greater_equal_bcast_expanded_function_O1 = opset16.Greater(x, y)
E       GreaterOrEqual_test_greater_equal_bcast_expanded_function_O2 = opset16.Equal(x, y)
E       greater_equal = opset16.Or(GreaterOrEqual_test_greater_equal_bcast_expanded_function_O1, GreaterOrEqual_test_greater_equal_bcast_expanded_function_O2)
E       return greater_equal

AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_scatternd_multiply' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_scatternd_multiply.py'))
----
import numpy
from onnx import TensorProto
from onnx.helper import make_tensor
from onnxscript import script, external_tensor
from onnxscript.values import Opset
from onnxscript.onnx_types import FLOAT, INT64
from onnxscript.onnx_opset import opset18

@script()
def bck_test_scatternd_multiply(data: FLOAT[4,4,4], indices: INT64[2,1], updates: FLOAT[2,4,4]) -> (FLOAT[4,4,4]):
    y = opset18.ScatterND(data, indices, updates, reduction='mul')
    return y
onnxscript\backend\onnx_export_test.py:116: in extract_functions
    mod = importlib.import_module(import_name)
C:\hostedtoolcache\windows\Python\3.8.10\x64\lib\importlib\__init__.py:127: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
E   ModuleNotFoundError: No module named 'onnxscript.tests.onnx_backend_test_code.test_scatternd_multiply'

The above exception was the direct cause of the following exception:
.nox\test\lib\site-packages\parameterized\parameterized.py:620: in standalone_func
    return func(*(a + p.args), **p.kwargs, **kw)
onnxscript\backend\onnx_export_test.py:235: in test_export2python_produces_correct_onnx_script_model
    functions = extract_functions(backend_test.name, code, self.test_folder)
onnxscript\backend\onnx_export_test.py:118: in extract_functions
    raise AssertionError(
E   AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_scatternd_multiply' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_scatternd_multiply.py'))
E   ----
E   import numpy
E   from onnx import TensorProto
E   from onnx.helper import make_tensor
E   from onnxscript import script, external_tensor
E   from onnxscript.values import Opset
E   from onnxscript.onnx_types import FLOAT, INT64
E   from onnxscript.onnx_opset import opset18
E   
E   @script()
E   def bck_test_scatternd_multiply(data: FLOAT[4,4,4], indices: INT64[2,1], updates: FLOAT[2,4,4]) -> (FLOAT[4,4,4]):
E       y = opset18.ScatterND(data, indices, updates, reduction='mul')
E       return y

AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_tril' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_tril.py'))
----
import numpy
from onnx import TensorProto
from onnx.helper import make_tensor
from onnxscript import script, external_tensor
from onnxscript.values import Opset
from onnxscript.onnx_types import INT64
from onnxscript.onnx_opset import opset14

@script()
def bck_test_tril(x: INT64[4,5]) -> (INT64[4,5]):
    y = opset14.Trilu(x, upper=0)
    return y
onnxscript\backend\onnx_export_test.py:116: in extract_functions
    mod = importlib.import_module(import_name)
C:\hostedtoolcache\windows\Python\3.8.10\x64\lib\importlib\__init__.py:127: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
E   ModuleNotFoundError: No module named 'onnxscript.tests.onnx_backend_test_code.test_tril'

The above exception was the direct cause of the following exception:
.nox\test\lib\site-packages\parameterized\parameterized.py:620: in standalone_func
    return func(*(a + p.args), **p.kwargs, **kw)
onnxscript\backend\onnx_export_test.py:235: in test_export2python_produces_correct_onnx_script_model
    functions = extract_functions(backend_test.name, code, self.test_folder)
onnxscript\backend\onnx_export_test.py:118: in extract_functions
    raise AssertionError(
E   AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_tril' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_tril.py'))
E   ----
E   import numpy
E   from onnx import TensorProto
E   from onnx.helper import make_tensor
E   from onnxscript import script, external_tensor
E   from onnxscript.values import Opset
E   from onnxscript.onnx_types import INT64
E   from onnxscript.onnx_opset import opset14
E   
E   @script()
E   def bck_test_tril(x: INT64[4,5]) -> (INT64[4,5]):
E       y = opset14.Trilu(x, upper=0)
E       return y

AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_roialign_mode_max' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_roialign_mode_max.py'))
----
import numpy
from onnx import TensorProto
from onnx.helper import make_tensor
from onnxscript import script, external_tensor
from onnxscript.values import Opset
from onnxscript.onnx_types import FLOAT, INT64
from onnxscript.onnx_opset import opset16

@script()
def bck_test_roialign_mode_max(X: FLOAT[1,1,10,10], rois: FLOAT[3,4], batch_indices: INT64[3]) -> (FLOAT[3,1,5,5]):
    Y = opset16.RoiAlign(X, rois, batch_indices, coordinate_transformation_mode='output_half_pixel', mode='max', output_height=5, output_width=5, sampling_ratio=2, spatial_scale=1.0)
    return Y
onnxscript\backend\onnx_export_test.py:116: in extract_functions
    mod = importlib.import_module(import_name)
C:\hostedtoolcache\windows\Python\3.8.10\x64\lib\importlib\__init__.py:127: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
E   ModuleNotFoundError: No module named 'onnxscript.tests.onnx_backend_test_code.test_roialign_mode_max'

The above exception was the direct cause of the following exception:
.nox\test\lib\site-packages\parameterized\parameterized.py:620: in standalone_func
    return func(*(a + p.args), **p.kwargs, **kw)
onnxscript\backend\onnx_export_test.py:235: in test_export2python_produces_correct_onnx_script_model
    functions = extract_functions(backend_test.name, code, self.test_folder)
onnxscript\backend\onnx_export_test.py:118: in extract_functions
    raise AssertionError(
E   AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_roialign_mode_max' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_roialign_mode_max.py'))
E   ----
E   import numpy
E   from onnx import TensorProto
E   from onnx.helper import make_tensor
E   from onnxscript import script, external_tensor
E   from onnxscript.values import Opset
E   from onnxscript.onnx_types import FLOAT, INT64
E   from onnxscript.onnx_opset import opset16
E   
E   @script()
E   def bck_test_roialign_mode_max(X: FLOAT[1,1,10,10], rois: FLOAT[3,4], batch_indices: INT64[3]) -> (FLOAT[3,1,5,5]):
E       Y = opset16.RoiAlign(X, rois, batch_indices, coordinate_transformation_mode='output_half_pixel', mode='max', output_height=5, output_width=5, sampling_ratio=2, spatial_scale=1.0)
E       return Y

AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_softmax_axis_2' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_softmax_axis_2.py'))
----
import numpy
from onnx import TensorProto
from onnx.helper import make_tensor
from onnxscript import script, external_tensor
from onnxscript.values import Opset
from onnxscript.onnx_types import FLOAT
from onnxscript.onnx_opset import opset13

@script()
def bck_test_softmax_axis_2(x: FLOAT[3,4,5]) -> (FLOAT[3,4,5]):
    y = opset13.Softmax(x, axis=2)
    return y
onnxscript\backend\onnx_export_test.py:116: in extract_functions
    mod = importlib.import_module(import_name)
C:\hostedtoolcache\windows\Python\3.8.10\x64\lib\importlib\__init__.py:127: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
E   ModuleNotFoundError: No module named 'onnxscript.tests.onnx_backend_test_code.test_softmax_axis_2'

The above exception was the direct cause of the following exception:
.nox\test\lib\site-packages\parameterized\parameterized.py:620: in standalone_func
    return func(*(a + p.args), **p.kwargs, **kw)
onnxscript\backend\onnx_export_test.py:235: in test_export2python_produces_correct_onnx_script_model
    functions = extract_functions(backend_test.name, code, self.test_folder)
onnxscript\backend\onnx_export_test.py:118: in extract_functions
    raise AssertionError(
E   AssertionError: Unable to import 'onnxscript.tests.onnx_backend_test_code.test_softmax_axis_2' (file: WindowsPath('D:/a/onnxscript/onnxscript/onnxscript/tests/onnx_backend_test_code/test_softmax_axis_2.py'))
E   ----
E   import numpy
E   from onnx import TensorProto
E   from onnx.helper import make_tensor
E   from onnxscript import script, external_tensor
E   from onnxscript.values import Opset
E   from onnxscript.onnx_types import FLOAT
E   from onnxscript.onnx_opset import opset13
E   
E   @script()
E   def bck_test_softmax_axis_2(x: FLOAT[3,4,5]) -> (FLOAT[3,4,5]):
E       y = opset13.Softmax(x, axis=2)
E       return y