fix prompt generator check

SciPhi-AI · Sep 20, 2023 · 8f25106 · 8f25106
1 parent c8dc8d1
commit 8f25106
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/.github/scripts/check_prompt_generators.py b/.github/scripts/check_prompt_generators.py
@@ -2,10 +2,8 @@
 import os
 
 from sciphi.config import DataConfig
-from sciphi.prompt import PromptGenerator, PromptManager
-from sciphi.core.utils import (
-    get_data_config_dir,
-)
+from sciphi.core.utils import get_data_config_dir
+from sciphi.prompt import PromptGenerator
 
 NUM_SAMPLES = 1_000_000
 if __name__ == "__main__":