Move

justinchuby · justinchuby · commit 56c1bba175da · 2024-08-30T22:57:37.000Z
diff --git a/onnxscript/ir/_schemas.py b/onnxscript/ir/_schemas.py
diff --git a/onnxscript/ir/_schemas_test.py b/onnxscript/ir/_schemas_test.py
@@ -10,7 +10,7 @@
 import onnxscript
 import onnxscript.testing
 from onnxscript import FLOAT, INT64, ir
-from onnxscript._internal import _schemas
+from onnxscript.ir import _schemas
 
 _TestTypeVarConstraints = TypeVar("_TestTypeVarConstraints", INT64, FLOAT)
 _TestTypeVarOneBound = TypeVar("_TestTypeVarOneBound", bound=INT64)
diff --git a/onnxscript/values.py b/onnxscript/values.py
@@ -24,7 +24,8 @@
 
 from onnxscript import converter as converter_module
 from onnxscript import irbuilder, sourceinfo, type_annotation
-from onnxscript._internal import _schemas, ast_utils, deprecation
+from onnxscript._internal import ast_utils, deprecation
+from onnxscript.ir import _schemas
 
 _ATTRIBUTE_TYPE_TO_PYTHON_TYPE = {
     onnx.defs.OpSchema.AttrType.FLOAT: float,
