Skip to content

Commit 0c90804

Browse files
committed
change back the defaults in test scripts
1 parent 1ac689a commit 0c90804

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

scripts/test_freeform_skills.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949

5050
ds = Dataset.from_list(samples)
5151

52-
skills_flow = SynthSkillsFlow(client, "mixtral", teacher_model, 30).get_flow()
52+
skills_flow = SynthSkillsFlow(client, "mixtral", teacher_model, 1).get_flow()
5353
skills_pipe = Pipeline(skills_flow)
5454

5555
sdg = SDG([skills_pipe])

scripts/test_grounded_skills.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@
9797

9898
ds = Dataset.from_list(samples)
9999

100-
skills_flow = SynthGroundedSkillsFlow(client, "mixtral", teacher_model, 30).get_flow()
100+
skills_flow = SynthGroundedSkillsFlow(client, "mixtral", teacher_model, 10).get_flow()
101101
skills_pipe = Pipeline(skills_flow)
102102

103103
sdg = SDG([skills_pipe])

scripts/test_knowledge.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,8 @@
3838

3939
ds = Dataset.from_list(samples)
4040

41-
mmlu_flow = MMLUBenchFlow(client, "mixtral", teacher_model, 30).get_flow()
42-
knowledge_flow = SynthKnowledgeFlow(client, "mixtral", teacher_model, 30).get_flow()
41+
mmlu_flow = MMLUBenchFlow(client, "mixtral", teacher_model, 1).get_flow()
42+
knowledge_flow = SynthKnowledgeFlow(client, "mixtral", teacher_model, 1).get_flow()
4343
knowledge_pipe = Pipeline(knowledge_flow)
4444
mmlu_pipe = Pipeline(mmlu_flow)
4545

0 commit comments

Comments
 (0)