From 213cfb626bd12de2df143a8c0196028e21ee1a87 Mon Sep 17 00:00:00 2001 From: Justin Chu Date: Mon, 16 Jan 2023 17:34:08 +0000 Subject: [PATCH] Fix init --- orttraining/orttraining/python/training/__init__.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/orttraining/orttraining/python/training/__init__.py b/orttraining/orttraining/python/training/__init__.py index ef785166a1a6a..3b28514819b2e 100644 --- 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