Add guard to hegel import

hegelai · Jul 6, 2023 · 99c8458 · 99c8458
1 parent c3acc3d
commit 99c8458
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/post-commit.yaml b/.github/workflows/post-commit.yaml
@@ -27,4 +27,4 @@ jobs:
  python setup.py develop
  - name: Run tests
  run: |
- DEBUG=1 HEGELAI_API_KEY=1 python examples/prompttests/example.py
+ DEBUG=1 python examples/prompttests/example.py
diff --git a/prompttools/experiment/openai_chat_experiment.py b/prompttools/experiment/openai_chat_experiment.py
@@ -12,7 +12,6 @@
 
 from prompttools.mock.mock import mock_chat_completion_fn
 from prompttools.experiment.experiment import Experiment
-from hegel.scribe import HegelScribe 
 
 
 class OpenAIChatExperiment(Experiment):
@@ -54,6 +53,7 @@ def __init__(
  ):
  self.use_scribe = use_scribe
  if use_scribe:
+ from hegel.scribe import HegelScribe 
  self.completion_fn = HegelScribe(
  name=scribe_name, completion_fn=openai.ChatCompletion.create
  )

diff --git a/prompttools/experiment/openai_completion_experiment.py b/prompttools/experiment/openai_completion_experiment.py
@@ -11,7 +11,6 @@
 
 from prompttools.mock.mock import mock_completion_fn
 from prompttools.experiment.experiment import Experiment
-from hegel.scribe import HegelScribe
 
 
 class OpenAICompletionExperiment(Experiment):
@@ -61,6 +60,7 @@ def __init__(
  ):
  self.use_scribe = use_scribe
  if use_scribe:
+ from hegel.scribe import HegelScribe
  self.completion_fn = HegelScribe(
  name=scribe_name, completion_fn=openai.Completion.create
  )