tidy-ups

flaxandteal · Feb 17, 2024 · fbab6f4 · fbab6f4
1 parent 3df11dc
commit fbab6f4
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/python/bonn/category_manager.py b/python/bonn/category_manager.py
@@ -124,7 +124,7 @@ def closest(self, text, cat, classifier_bow_vec):
         ]
 
     def strip_document(self, doc):
-        if type(doc) is list:
+        if isinstance(doc, list):
             doc = " ".join(doc)
 
         docs = doc.split(",")

diff --git a/tests/category_manager_test.py b/tests/category_manager_test.py
@@ -1,8 +1,6 @@
 import pytest
 from nltk import download
 from pathlib import Path
-from fastapi.testclient import TestClient
-from unittest.mock import MagicMock
 from bonn.extract import CategoryManager, FfModel
 from dynaconf import Dynaconf