From befc4593c3c9e082f1690c2f7509dc6a47ecf3aa Mon Sep 17 00:00:00 2001
From: ylzz1997 <ylzz1997@outlook.com>
Date: Sat, 22 Jul 2023 23:02:52 +0800
Subject: [PATCH] ruff fix

---
 preprocess_hubert_f0.py | 2 +-
 train_diff.py           | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/preprocess_hubert_f0.py b/preprocess_hubert_f0.py
index ec6dcc3..b5d1f76 100644
--- a/preprocess_hubert_f0.py
+++ b/preprocess_hubert_f0.py
@@ -5,12 +5,12 @@ import random
 from concurrent.futures import ProcessPoolExecutor
 from glob import glob
 from random import shuffle
-from loguru import logger
 
 import librosa
 import numpy as np
 import torch
 import torch.multiprocessing as mp
+from loguru import logger
 from tqdm import tqdm
 
 import diffusion.logger.utils as du
diff --git a/train_diff.py b/train_diff.py
index 4cdc0eb..65ba382 100644
--- a/train_diff.py
+++ b/train_diff.py
@@ -1,6 +1,7 @@
 import argparse
 
 import torch
+from loguru import logger
 from torch.optim import lr_scheduler
 
 from diffusion.data_loaders import get_data_loaders
@@ -8,7 +9,7 @@ from diffusion.logger import utils
 from diffusion.solver import train
 from diffusion.unit2mel import Unit2Mel
 from diffusion.vocoder import Vocoder
-from loguru import logger
+
 
 def parse_args(args=None, namespace=None):
     """Parse command-line arguments."""