diff --git a/libs/langchain/tests/unit_tests/conftest.py b/libs/langchain/tests/unit_tests/conftest.py index ac58a15c41a..d347eb15cf7 100644 --- a/libs/langchain/tests/unit_tests/conftest.py +++ b/libs/langchain/tests/unit_tests/conftest.py @@ -49,10 +49,10 @@ def pytest_collection_modifyitems( # Used to avoid repeated calls to `util.find_spec` required_pkgs_info: dict[str, bool] = {} - only_extended = config.getoption("--only-extended") or False - only_core = config.getoption("--only-core") or False + only_extended = config.getoption("--only-extended", default=False) + only_core = config.getoption("--only-core", default=False) - if not config.getoption("--community"): + if not config.getoption("--community", default=False): skip_community = pytest.mark.skip(reason="need --community option to run") for item in items: if "community" in item.keywords: diff --git a/libs/langchain_v1/tests/unit_tests/conftest.py b/libs/langchain_v1/tests/unit_tests/conftest.py index ebc0b8376e3..0b38ae788f7 100644 --- a/libs/langchain_v1/tests/unit_tests/conftest.py +++ b/libs/langchain_v1/tests/unit_tests/conftest.py @@ -80,8 +80,8 @@ def pytest_collection_modifyitems(config: pytest.Config, items: Sequence[pytest. # Used to avoid repeated calls to `util.find_spec` required_pkgs_info: dict[str, bool] = {} - only_extended = config.getoption("--only-extended") or False - only_core = config.getoption("--only-core") or False + only_extended = config.getoption("--only-extended", default=False) + only_core = config.getoption("--only-core", default=False) if only_extended and only_core: msg = "Cannot specify both `--only-extended` and `--only-core`."