github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-17 07:26:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

infra: add print rule to ruff (#16221)

Browse Source 
Added noqa for existing prints. Can slowly remove / will prevent more
being intro'd
This commit is contained in:
Erick Friis
2024-02-09 16:13:30 -08:00
committed by GitHub
parent c07c0da01a
commit 3a2eb6e12b
246 changed files with 563 additions and 470 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
libs/community/tests/unit_tests/llms/test_cerebriumai.py
View File

@@ -1,6 +1,5 @@
"""Test CerebriumAI llm"""
from langchain_core.pydantic_v1 import SecretStr
from pytest import CaptureFixture, MonkeyPatch
@@ -14,7 +13,7 @@ def test_api_key_is_secret_string() -> None:
def test_api_key_masked_when_passed_via_constructor(capsys: CaptureFixture) -> None:
llm = CerebriumAI(cerebriumai_api_key="secret-api-key")
print(llm.cerebriumai_api_key, end="")
print(llm.cerebriumai_api_key, end="") # noqa: T201
captured = capsys.readouterr()
assert captured.out == "**********"
@@ -26,7 +25,7 @@ def test_api_key_masked_when_passed_from_env(
) -> None:
monkeypatch.setenv("CEREBRIUMAI_API_KEY", "secret-api-key")
llm = CerebriumAI()
print(llm.cerebriumai_api_key, end="")
print(llm.cerebriumai_api_key, end="") # noqa: T201
captured = capsys.readouterr()
assert captured.out == "**********"