Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
Updates
Browse files Browse the repository at this point in the history
  • Loading branch information
ethanwharris committed Oct 25, 2021
1 parent 41ab62a commit 54ece60
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion flash/text/classification/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ def _transform_label(label_to_class_mapping: Dict[str, int], target: str, ex: Di

@staticmethod
def _multilabel_target(targets: List[str], element: Dict[str, Any]) -> Dict[str, Any]:
targets = [int(element.pop(target)) for target in targets]
targets = [element.pop(target) for target in targets]
element[DefaultDataKeys.TARGET] = targets
return element

Expand Down

0 comments on commit 54ece60

Please sign in to comment.