Skip to content

Commit

Permalink
Add additional possibility of different parsing, but there's nothing …
Browse files Browse the repository at this point in the history
…else to add for the moment
  • Loading branch information
plaguss committed Oct 7, 2024
1 parent e027f99 commit 75350de
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/distilabel/steps/tasks/text_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -336,6 +336,8 @@ def format_output(
) -> Dict[str, Any]:
"""The output is formatted as a dictionary with the `generation`. The `model_name`
will be automatically included within the `process` method of `Task`."""
if self.use_default_structured_output:
return self._format_structured_output(output)
return self._format_structured_output(output)

@override
Expand Down

0 comments on commit 75350de

Please sign in to comment.