Autopep

ucsinfer/__main__.py

@@ -107,7 +107,7 @@ def evaluate(dataset, offset, limit):
 
     miss_counts = []
     for cat in cats:
-        miss_counts.append((cat, len([x for x in results \
+        miss_counts.append((cat, len([x for x in results
                                       if x['catid'] == cat and x['result'] == 'MISS'])))
 
     miss_counts = sorted(miss_counts, key=lambda x: x[1])

ucsinfer/inference.py

@@ -11,6 +11,7 @@ import platformdirs
 
 from sentence_transformers import SentenceTransformer
 
+
 def classify_text_ranked(text, embeddings_list, model, limit=5):
     text_embedding = model.encode(text, convert_to_numpy=True)
     embeddings = np.array([info['Embedding'] for info in embeddings_list])
@@ -32,6 +33,7 @@ class Ucs(NamedTuple):
                    subcategory=d['SubCategory'],
                    explanations=d['Explanations'], synonymns=d['Synonyms'])
 
+
 def load_ucs() -> list[Ucs]:
     FILE_ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
     cats = []
@@ -43,6 +45,7 @@ def load_ucs() -> list[Ucs]:
 
     return [Ucs.from_dict(cat) for cat in cats]
 
+
 class InferenceContext:
     """
     Maintains caches and resources for UCS category inference.
@@ -108,9 +111,8 @@ class InferenceContext:
         :raises: StopIterator if CatId is not on the schedule 
         """
         i = (
-                (x.category, x.subcategory, x.explanations) \
+            (x.category, x.subcategory, x.explanations)
             for x in self.catlist if x.catid == catid
         )
 
         return next(i)
-

ucsinfer/util.py

@@ -6,6 +6,7 @@ from re import match
 
 from .inference import Ucs
 
+
 def ffmpeg_description(path: str) -> Optional[str]:
     result = subprocess.run(['ffprobe', '-show_format', '-of',
                              'json', path], capture_output=True)