Skip to content

Commit df4aab3

Browse files
committed
Fix ruff import ordering and Black formatting errors
1 parent 7397c78 commit df4aab3

File tree

3 files changed

+12
-7
lines changed

3 files changed

+12
-7
lines changed

policylens/apps/claims/ml/contracts.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,8 @@
1414
from __future__ import annotations
1515

1616
import hashlib
17-
from typing import Final, Sequence
17+
from collections.abc import Sequence
18+
from typing import Final
1819

1920
# Contract version should bump only when FEATURE_NAMES meaning or order changes.
2021
FEATURE_CONTRACT_VERSION: Final[str] = "v1"
@@ -28,14 +29,12 @@
2829
"priority_is_low",
2930
"summary_length",
3031
"summary_has_digits",
31-
3232
# Document metadata
3333
"documents_count",
3434
"documents_total_bytes",
3535
"documents_has_pdf",
3636
"documents_has_image",
3737
"documents_has_text",
38-
3938
# Checklist completeness (if present)
4039
"checklist_required_count",
4140
"checklist_satisfied_count",

policylens/apps/claims/ml/features.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,9 @@ def extract_features(*, claim: Claim) -> FeatureResult:
5656
documents_count = float(docs.count())
5757
documents_total_bytes = float(sum((d.size_bytes or 0) for d in docs))
5858
documents_has_pdf = any((d.content_type or "").lower() == "application/pdf" for d in docs)
59-
documents_has_image = any((d.content_type or "").lower() in {"image/jpeg", "image/png"} for d in docs)
59+
documents_has_image = any(
60+
(d.content_type or "").lower() in {"image/jpeg", "image/png"} for d in docs
61+
)
6062
documents_has_text = any((d.content_type or "").lower() == "text/plain" for d in docs)
6163

6264
checklist = ChecklistItem.objects.filter(claim=claim)

tests/test_ml_feature_contract.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
from policylens.apps.claims.ml.contracts import FEATURE_NAMES, feature_contract_hash
1414
from policylens.apps.claims.ml.features import extract_features
15-
from policylens.apps.claims.models import Claim, ClaimDocument, ChecklistItem
15+
from policylens.apps.claims.models import ChecklistItem, Claim, ClaimDocument
1616
from tests.factories import PolicyFactory
1717

1818

@@ -40,8 +40,12 @@ def test_extract_features_returns_values_aligned_to_feature_names(settings, tmp_
4040
created_by="tester",
4141
)
4242

43-
ChecklistItem.objects.create(claim=claim, key="id", label="ID", is_required=True, is_satisfied=False)
44-
ChecklistItem.objects.create(claim=claim, key="addr", label="Address", is_required=True, is_satisfied=True)
43+
ChecklistItem.objects.create(
44+
claim=claim, key="id", label="ID", is_required=True, is_satisfied=False
45+
)
46+
ChecklistItem.objects.create(
47+
claim=claim, key="addr", label="Address", is_required=True, is_satisfied=True
48+
)
4549

4650
uploaded = SimpleUploadedFile("doc.pdf", b"pdfdata", content_type="application/pdf")
4751
ClaimDocument.objects.create(

0 commit comments

Comments
 (0)