Remove code duplication - #1413

stanfordnlp · Aug 25, 2024 · 3de69ca · 3de69ca
1 parent 33a5256
commit 3de69ca
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/stanza/pipeline/pos_processor.py b/stanza/pipeline/pos_processor.py
@@ -79,13 +79,10 @@ def process(self, document):
         idx = []
         with torch.no_grad():
             if self._tqdm:
-                for i, b in enumerate(tqdm(batch)):
-                    idx.extend(b[-1])
-                    preds += self.trainer.predict(b)
-            else:
-                for i, b in enumerate(batch):
-                    idx.extend(b[-1])
-                    preds += self.trainer.predict(b)
+                batch = tqdm(batch)
+            for i, b in enumerate(batch):
+                idx.extend(b[-1])
+                preds += self.trainer.predict(b)
 
         preds = unsort(preds, idx)
         dataset.doc.set([doc.UPOS, doc.XPOS, doc.FEATS], [y for x in preds for y in x])