lint

Signed-off-by: Justin Chu <[email protected]>

Author: justinchuby

onnxscript/_internal/converter.py

@@ -27,9 +27,11 @@
     autocast,
     irbuilder,
     param_manipulation,
-    type_annotation as ta,
     values,
 )
+from onnxscript._internal import (
+    type_annotation as ta,
+)
 
 logger = logging.getLogger("onnxscript")
 

onnxscript/_internal/values.py

@@ -30,7 +30,8 @@
 from typing_extensions import ParamSpec
 
 from onnxscript import sourceinfo
-from onnxscript._internal import ast_utils, converter as converter_module, deprecation, irbuilder, type_annotation
+from onnxscript._internal import ast_utils, deprecation, irbuilder, type_annotation
+from onnxscript._internal import converter as converter_module
 from onnxscript.ir import _schemas
 from onnxscript.onnx_types import ONNXType
 
@@ -577,7 +578,9 @@ def __getitem__(self, instance):
 
         def fun(*args, **kwargs):
             # FIXME(after #225): Move import to the top of the file.
-            from onnxscript._internal import evaluator  # pylint: disable=import-outside-toplevel
+            from onnxscript._internal import (
+                evaluator,  # pylint: disable=import-outside-toplevel
+            )
 
             with evaluator.default_as(instance):
                 return self.__call__(*args, **kwargs)

onnxscript/evaluator.py

@@ -22,4 +22,5 @@
     "OnnxReferenceRuntimeEvaluator",
     "default",
     "default_as",
+    "set_default",
 ]