format code

source/services/ner/train/train.py

@@ -10,9 +10,12 @@ from huggingface_hub import notebook_login
 notebook_login()
 
 """
+import torch
 import datasets
 import evaluate
 import numpy as np
+from tqdm.auto import tqdm
+
 from transformers import Trainer, AutoModelForTokenClassification, TrainingArguments, DataCollatorForTokenClassification
 
 dataset = datasets.load_dataset("json", data_files="data/ner_input_data/ner_dataset.json")
@@ -283,8 +286,6 @@ def postprocess(predictions, labels):
     ]
     return true_labels, true_predictions
 
-from tqdm.auto import tqdm
-import torch
 
 progress_bar = tqdm(range(num_training_steps))