Formatting isort

Author: sadda

wildlife_tools/data/dataset.py

@@ -1,13 +1,13 @@
 import json
 import os
 import pickle
-import torch
 from typing import Callable, List
 
 import cv2
 import numpy as np
 import pandas as pd
 import pycocotools.mask as mask_coco
+import torch
 from PIL import Image
 
 

wildlife_tools/features/deep.py

@@ -1,7 +1,9 @@
+from typing import Optional
+
 import torch
 from tqdm import tqdm
 from transformers import CLIPModel, CLIPProcessor
-from typing import Optional
+
 from ..data import FeatureDataset, ImageDataset
 from ..tools import check_dataset_output
 

wildlife_tools/features/memory.py

@@ -1,4 +1,5 @@
 from tqdm import tqdm
+
 from ..data import FeatureDataset, ImageDataset
 from ..tools import check_dataset_output
 

wildlife_tools/inference/classifier.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
+from typing import Tuple
+
 import numpy as np
 import pandas as pd
 import torch
-from typing import Tuple
 
 
 class KnnClassifier:

wildlife_tools/train/callbacks.py

@@ -1,6 +1,8 @@
 import os
-from torch.utils.tensorboard import SummaryWriter
 from typing import List
+
+from torch.utils.tensorboard import SummaryWriter
+
 from .trainer import BasicTrainer
 
 

wildlife_tools/train/trainer.py

@@ -1,10 +1,12 @@
 import os
 import random
+from typing import Callable, Optional
+
 import numpy as np
 import torch
 import torch.backends.cudnn
 from tqdm import tqdm
-from typing import Optional, Callable
+
 from ..data import ImageDataset
 from ..tools import check_dataset_output