Skip to content

Commit

Permalink
Merge pull request #150 from parlance/feature/labels_list
Browse files Browse the repository at this point in the history
Ensure labels are a list before calling the c++ api
  • Loading branch information
Sean Naren authored Jun 7, 2020
2 parents 3268f58 + 7619d2c commit 4d3d052
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ctcdecode/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ def __init__(self, labels, model_path=None, alpha=0, beta=0, cutoff_top_n=40, cu
self._beam_width = beam_width
self._scorer = None
self._num_processes = num_processes
self._labels = labels
self._labels = list(labels) # Ensure labels are a list
self._num_labels = len(labels)
self._blank_id = blank_id
self._log_probs = 1 if log_probs_input else 0
Expand Down

0 comments on commit 4d3d052

Please sign in to comment.