leavoigt commited on
Commit
fa5983b
1 Parent(s): 6fef1b7

Update utils/target_classifier.py

Browse files
Files changed (1) hide show
  1. utils/target_classifier.py +1 -1
utils/target_classifier.py CHANGED
@@ -23,7 +23,7 @@ def get_target_labels(preds):
23
  """
24
 
25
  # Turn into list
26
- preds_list = list(preds)
27
 
28
  # Get label names
29
  predictions_names=[]
 
23
  """
24
 
25
  # Turn into list
26
+ preds_list = preds.to_list()
27
 
28
  # Get label names
29
  predictions_names=[]