Skip to content

Commit

Permalink
Turn print statement into log line
Browse files Browse the repository at this point in the history
  • Loading branch information
AngledLuffa committed Jan 13, 2024
1 parent 6e12731 commit 0465dc6
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion stanza/models/lemma_classifier/utils.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
from collections import Counter, defaultdict
import logging
import os
from typing import List, Tuple, Any, Mapping

Expand Down Expand Up @@ -39,7 +40,7 @@ def load_dataset(data_path: str, batch_size=DEFAULT_BATCH_SIZE, get_counts: bool
# but we can still use those labels in a confusion matrix
label_decoder = dict(label_decoder)

print(label_decoder, f"Should be strings to ints")
logging.debug("Final label decoder: %s Should be strings to ints", label_decoder)

with open(data_path, "r+", encoding="utf-8") as f:
sentences, indices, labels, upos_ids, counts, upos_to_id = [], [], [], [], Counter(), defaultdict(str)
Expand Down

0 comments on commit 0465dc6

Please sign in to comment.