From d5fc59da742b3c671ae34952e94bb96286a0c8e7 Mon Sep 17 00:00:00 2001 From: shiv Date: Wed, 3 Jul 2024 03:49:52 +0000 Subject: [PATCH] :white_check_mark: updating test script imports Signed-off-by: shiv --- scripts/test_freeform_skills.py | 7 ++++--- scripts/test_grounded_skills.py | 7 ++++--- scripts/test_knowledge.py | 9 +++++---- 3 files changed, 13 insertions(+), 10 deletions(-) diff --git a/scripts/test_freeform_skills.py b/scripts/test_freeform_skills.py index a8612c09..13ae235a 100644 --- a/scripts/test_freeform_skills.py +++ b/scripts/test_freeform_skills.py @@ -3,9 +3,10 @@ from openai import OpenAI # First Party -from src.instructlab.sdg import SDG -from src.instructlab.sdg.default_flows import SynthSkillsFlow -from src.instructlab.sdg.pipeline import Pipeline +from instructlab.sdg import SDG +from instructlab.sdg.default_flows import SynthSkillsFlow +from instructlab.sdg.pipeline import Pipeline + # for vLLM endpoints, the api_key remains "EMPTY" openai_api_key = "EMPTY" diff --git a/scripts/test_grounded_skills.py b/scripts/test_grounded_skills.py index 338edb6c..f0e3f1a9 100644 --- a/scripts/test_grounded_skills.py +++ b/scripts/test_grounded_skills.py @@ -3,9 +3,10 @@ from openai import OpenAI # First Party -from src.instructlab.sdg import SDG -from src.instructlab.sdg.default_flows import SynthGroundedSkillsFlow -from src.instructlab.sdg.pipeline import Pipeline +from instructlab.sdg import SDG +from instructlab.sdg.default_flows import SynthGroundedSkillsFlow +from instructlab.sdg.pipeline import Pipeline + # for vLLM endpoints, the api_key remains "EMPTY" openai_api_key = "EMPTY" diff --git a/scripts/test_knowledge.py b/scripts/test_knowledge.py index aeedcf59..05d01c70 100644 --- a/scripts/test_knowledge.py +++ b/scripts/test_knowledge.py @@ -6,11 +6,12 @@ from openai import OpenAI # First Party -from src.instructlab.sdg import SDG -from src.instructlab.sdg.default_flows import MMLUBenchFlow, SynthKnowledgeFlow -from src.instructlab.sdg.pipeline import Pipeline +from instructlab.sdg import SDG +from instructlab.sdg.default_flows import MMLUBenchFlow, SynthKnowledgeFlow +from instructlab.sdg.pipeline import Pipeline -# Please don't add you vLLM endpoint key here + +# for vLLM endpoints, the api_key remains "EMPTY" openai_api_key = "EMPTY" openai_api_base = "Add model endpoint here"