organize import

Author: Eleven1Liu

libmultilabel/nn/model.py

@@ -1,13 +1,12 @@
 from abc import abstractmethod
 
-import numpy as np
-
 import lightning as L
+import numpy as np
 import torch
 import torch.nn.functional as F
 import torch.optim as optim
 
-from ..common_utils import dump_log, argsort_top_k
+from ..common_utils import argsort_top_k, dump_log
 from ..nn.metrics import get_metrics, tabulate_metrics
 
 

search_params.py

@@ -5,16 +5,15 @@
 from datetime import datetime
 from pathlib import Path
 
-import yaml
+import numpy as np
 import ray.train as ray_train
+import yaml
 from ray import tune
 from ray.tune.schedulers import ASHAScheduler
 
-import numpy as np
-
+from libmultilabel.common_utils import AttributeDict, timer
 from libmultilabel.nn import data_utils
 from libmultilabel.nn.nn_utils import set_seed
-from libmultilabel.common_utils import AttributeDict, timer
 from torch_trainer import TorchTrainer
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s:%(message)s")