✅ updating test script imports

Signed-off-by: shiv <[email protected]>
instructlab · Jul 3, 2024 · d5fc59d · d5fc59d
1 parent bada90e
commit d5fc59d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 10 deletions.
diff --git 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
@@ -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
@@ -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"