Skip to content

Commit

Permalink
Consolidate langchain fixtures, fix conda command
Browse files Browse the repository at this point in the history
  • Loading branch information
dagardner-nv committed Oct 16, 2024
1 parent ea6b154 commit 1a1c194
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions tests/morpheus_llm/stages/arxiv/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,10 +20,10 @@

from _utils import import_or_skip

SKIP_REASON = ("Tests for the arxiv_source require a number of packages not installed in the Morpheus development "
"environment. To install these run:\n"
"`conda env update --solver=libmamba -n morpheus "
"--file conda/environments/dev_cuda-125_arch-x86_64.yaml --prune`")
SKIP_REASON = (
"Tests for the arxiv_source require a number of packages not installed in the Morpheus development "
"environment. To install these run:\n"
"`conda env update --solver=libmamba -n morpheus --file conda/environments/examples_cuda-125_arch-x86_64.yaml`")


@pytest.fixture(name="arxiv", autouse=True, scope='session')
Expand All @@ -35,19 +35,19 @@ def arxiv_fixture(fail_missing: bool):


@pytest.fixture(name="langchain", autouse=True, scope='session')
def langchain_fixture(fail_missing: bool):
def langchain_fixture(langchain):
"""
All of the tests in this subdir require langchain
"""
yield import_or_skip("langchain", reason=SKIP_REASON, fail_missing=fail_missing)
yield langchain


@pytest.fixture(name="langchain_community", autouse=True, scope='session')
def langchain_community_fixture(fail_missing: bool):
def langchain_community_fixture(langchain_community):
"""
All of the tests in this subdir require langchain_community
"""
yield import_or_skip("langchain_community", reason=SKIP_REASON, fail_missing=fail_missing)
yield langchain_community


@pytest.fixture(name="pypdf", autouse=True, scope='session')
Expand Down

0 comments on commit 1a1c194

Please sign in to comment.