qol: linting

src/interface/clip.py

@@ -1,7 +1,8 @@
 """CLIP interface module"""
 
 # libraries
-from typing import List, Dict, Union
+from typing import Dict, List, Union
+
 from PIL import Image
 
 # modules

src/model/clip.py

@@ -1,13 +1,15 @@
 """CLIP model for zero-shot classification; running on CPU machine"""
-from typing import List, Dict
-from PIL import Image
-import torch
+from typing import Dict, List
+
 import open_clip
+import torch
 from open_clip import tokenizer
+from PIL import Image
+
+from src.core.logger import logger
 
 # modules
 from src.core.singleton import SingletonMeta
-from src.core.logger import logger
 
 
 class ClipModel(metaclass=SingletonMeta):
@@ -24,14 +26,14 @@ class ClipModel(metaclass=SingletonMeta):
         jit: bool = False,
     ):
         logger.debug("creating CLIP Model Object")
-        self.config = dict(
-            model_name=model_name,
-            pretrained=pretrained,
-            precision="bf16",
-            device=torch.device("cpu"),
-            jit=jit,
-            cache_dir="model_dir/",
-        )
+        self.config = {
+            "model_name": model_name,
+            "pretrained": pretrained,
+            "precision": "bf16",
+            "device": "cpu",
+            "jit": jit,
+            "cache_dir": "model_dir/",
+        }
         self.model, self.preprocess = open_clip.create_model_from_pretrained(
             **self.config
         )