black

amaiya · Jun 14, 2024 · de5f383 · de5f383
1 parent 983fbe7
commit de5f383
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ktrain/text/eda.py b/ktrain/text/eda.py
@@ -369,9 +369,10 @@ def filter(self, obj):
                 "Length of obj is not consistent with the number of documents "
                 + "supplied to get_topic_model"
             )
-        #obj = np.array(obj) if isinstance(obj, list) else obj
-        #return obj[self.bool_array]
+        # obj = np.array(obj) if isinstance(obj, list) else obj
+        # return obj[self.bool_array]
         from itertools import compress
+
         return list(compress(obj, self.bool_array))
 
     def get_docs(self, topic_ids=[], doc_ids=[], rank=False):