vulnerability

Sleeping

leavoigt commited on Jan 5

Commit

fa5983b

•

1 Parent(s): 6fef1b7

Update utils/target_classifier.py

Files changed (1) hide show

utils/target_classifier.py CHANGED Viewed

@@ -23,7 +23,7 @@ def get_target_labels(preds):
     """
     # Turn into list
-    preds_list = list(preds)
     # Get label names
     predictions_names=[]

     """
     # Turn into list
+    preds_list = preds.to_list()
     # Get label names
     predictions_names=[]