Skip to content

Commit 4c3ce98

Browse files
committed
style: remove unused imports (pass ruff)
- tests/providers/test_anthropic_cache_policy.py: CacheMarkerPolicy was imported but never referenced (custom policies in tests use ad-hoc duck-typed classes, not the Protocol) - tests/checkpointer/test_postgres.py: model classes now used in test_models_import via __tablename__ assertions for substantive checks
1 parent ba2d1ca commit 4c3ce98

2 files changed

Lines changed: 4 additions & 1 deletion

File tree

tests/checkpointer/test_postgres.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,10 @@ def test_models_import() -> None:
1515
)
1616
assert EXPECTED_SCHEMA_VERSION == 1
1717
assert PARTITION_COUNT == 64
18+
# All three model classes are reachable via the public model module
19+
assert CubepiThread.__tablename__ == "cubepi_threads"
20+
assert CubepiMessage.__tablename__ == "cubepi_messages"
21+
assert CubepiSchemaVersion.__tablename__ == "cubepi_schema_version"
1822
# All three tables registered on cubepi_metadata
1923
assert "cubepi_threads" in cubepi_metadata.tables
2024
assert "cubepi_messages" in cubepi_metadata.tables

tests/providers/test_anthropic_cache_policy.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
from cubepi.providers.anthropic import (
66
AnthropicProvider,
7-
CacheMarkerPolicy,
87
DefaultCacheMarkerPolicy,
98
)
109
from cubepi.providers.base import Message, TextContent, UserMessage

0 commit comments

Comments
 (0)