Fix init

microsoft · Jan 16, 2023 · 213cfb6 · 213cfb6
1 parent d1e55a4
commit 213cfb6
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/orttraining/orttraining/python/training/__init__.py b/orttraining/orttraining/python/training/__init__.py
@@ -2,15 +2,16 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 # --------------------------------------------------------------------------
-
+# isort:skip_file
 from onnxruntime.capi._pybind_state import PropagateCastOpsStrategy, TrainingParameters
 from onnxruntime.capi.training.training_session import TrainingSession
 
-from . import amp, checkpoint, model_desc_validation, optim
-from .orttrainer import ORTTrainer, TrainStepInfo
-
 # Options need to be imported before `ORTTrainer`.
 from .orttrainer_options import ORTTrainerOptions
+from .orttrainer import ORTTrainer, TrainStepInfo
+
+from . import amp, checkpoint, model_desc_validation, optim
+
 
 try:
     from .ortmodule import ORTModule