Skip to content

Commit

Permalink
optimized: task and datasource modules
Browse files Browse the repository at this point in the history
  • Loading branch information
chakravarthik27 committed Oct 25, 2023
1 parent 76a4ea6 commit 6010c76
Show file tree
Hide file tree
Showing 2 changed files with 124 additions and 107 deletions.
13 changes: 0 additions & 13 deletions langtest/datahandler/datasource.py
Original file line number Diff line number Diff line change
Expand Up @@ -1355,19 +1355,6 @@ def load_data(self, *args, **kwargs) -> List[Sample]:
)
data.append(sample)

# elif self.task == "stereoset":
# data.append(
# StereoSetSample(
# test_type=item["type"],
# target=item["target"],
# bias_type=item["bias_type"],
# context=item["context"],
# sent_stereo=item["stereotype"],
# sent_antistereo=item["anti-stereotype"],
# sent_unrelated=item["unrelated"],
# )
# )

return data

def export_data(self, data: List[Sample], output_path: str):
Expand Down
Loading

0 comments on commit 6010c76

Please sign in to comment.