From 0cea7f456b4e7472d529c8f4b513dc1a22dcafab Mon Sep 17 00:00:00 2001 From: gsheni Date: Sat, 30 Mar 2024 16:52:13 -0400 Subject: [PATCH] fix install --- .github/workflows/install.yaml | 5 ++++- trane/llm/helpers.py | 1 - 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/.github/workflows/install.yaml b/.github/workflows/install.yaml index dc7d14e..5932762 100644 --- a/.github/workflows/install.yaml +++ b/.github/workflows/install.yaml @@ -33,4 +33,7 @@ jobs: python -m pip check - name: Install add ons run: | - python -m pip install "unpacked_sdist/[llm]" \ No newline at end of file + python -m pip install "unpacked_sdist/[llm]" + - name: Test by importing packages + run: | + python -c "import openai" \ No newline at end of file diff --git a/trane/llm/helpers.py b/trane/llm/helpers.py index b5a82eb..999b6e1 100644 --- a/trane/llm/helpers.py +++ b/trane/llm/helpers.py @@ -6,7 +6,6 @@ openai = import_or_none("openai") tiktoken = import_or_none("tiktoken") -anthropic = import_or_none("anthropic") ipython = import_or_none("IPython")