diff --git a/libs/community/tests/unit_tests/chat_models/test_snowflake.py b/libs/community/tests/unit_tests/chat_models/test_snowflake.py index 16e5e2cc83..2013b041d7 100644 --- a/libs/community/tests/unit_tests/chat_models/test_snowflake.py +++ b/libs/community/tests/unit_tests/chat_models/test_snowflake.py @@ -141,15 +141,18 @@ def test_session_handled_outside_not_closed() -> None: mock_del.assert_called_once() sp_session_mock.close.assert_not_called() + @pytest.mark.parametrize( "set_env_vars", - [{ - "temp_env_vars": { - "SNOWFLAKE_USERNAME": "user", - "SNOWFLAKE_ACCOUNT": "account", - "SNOWFLAKE_PASSWORD": "password", + [ + { + "temp_env_vars": { + "SNOWFLAKE_USERNAME": "user", + "SNOWFLAKE_ACCOUNT": "account", + "SNOWFLAKE_PASSWORD": "password", + } } - }], + ], indirect=["set_env_vars"], ) def test_del_called_without_session_provided(set_env_vars: Any) -> None: