From ca23ca510aebd124e9ffdbf69d54154ad301301b Mon Sep 17 00:00:00 2001 From: Im-Himanshu Date: Mon, 8 Jul 2024 13:14:31 +0530 Subject: [PATCH] isort the new files --- yolox/core/trainer.py | 2 +- yolox/utils/__init__.py | 2 +- yolox/utils/mlflow_logger.py | 12 +++++++----- 3 files changed, 9 insertions(+), 7 deletions(-) diff --git a/yolox/core/trainer.py b/yolox/core/trainer.py index 41fecc172..8f8016e57 100644 --- a/yolox/core/trainer.py +++ b/yolox/core/trainer.py @@ -14,9 +14,9 @@ from yolox.exp import Exp from yolox.utils import ( MeterBuffer, + MlflowLogger, ModelEMA, WandbLogger, - MlflowLogger, adjust_status, all_reduce_norm, get_local_rank, diff --git a/yolox/utils/__init__.py b/yolox/utils/__init__.py index 4e00bd3de..8365a6d18 100644 --- a/yolox/utils/__init__.py +++ b/yolox/utils/__init__.py @@ -9,8 +9,8 @@ from .dist import * from .ema import * from .logger import WandbLogger, setup_logger -from .mlflow_logger import MlflowLogger from .lr_scheduler import LRScheduler +from .mlflow_logger import MlflowLogger from .metric import * from .model_utils import * from .setup_env import * diff --git a/yolox/utils/mlflow_logger.py b/yolox/utils/mlflow_logger.py index 28b031822..69c1acfb1 100644 --- a/yolox/utils/mlflow_logger.py +++ b/yolox/utils/mlflow_logger.py @@ -13,16 +13,18 @@ For more information, please refer to: https://mlflow.org/docs/latest/introduction/index.html """ +from loguru import logger + +import torch + +from yolox.utils import is_main_process + import importlib.metadata import importlib.util import json import os -from collections.abc import MutableMapping import packaging.version - -from loguru import logger -import torch -from yolox.utils import is_main_process +from collections.abc import MutableMapping class MlflowLogger: