We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 12ec1d4 commit a9b7da5Copy full SHA for a9b7da5
1 file changed
tests/data/test_data_handler.py
@@ -250,12 +250,10 @@ def test_few_shot_split(dataset):
250
251
252
def _make_multiclass_mapping_with_oos(*, with_validation: bool) -> dict:
253
- in_domain = []
254
# Ensure enough samples per class so stratified splitting doesn't fail.
255
- for i in range(50):
256
- in_domain.append({"utterance": f"c0_{i}", "label": 0})
257
258
- in_domain.append({"utterance": f"c1_{i}", "label": 1})
+ in_domain = [{"utterance": f"c0_{i}", "label": 0} for i in range(50)] + [
+ {"utterance": f"c1_{i}", "label": 1} for i in range(50)
+ ]
259
260
oos = [{"utterance": f"oos_{i}"} for i in range(20)]
261
0 commit comments