Skip to content

Commit 0e59bbf

Browse files
committed
formatting fixes
Signed-off-by: phernandez <paul@basicmachines.co>
1 parent 3130e5a commit 0e59bbf

8 files changed

Lines changed: 45 additions & 26 deletions

File tree

benchmarks/src/basic_memory_benchmarks/cli.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,9 @@ def convert_structure_corpus(
153153
max_workers=max_workers,
154154
)
155155

156-
console.print(f"Structured ([green]{mode}[/green]): [cyan]{output_dir}[/cyan] ({doc_count} docs)")
156+
console.print(
157+
f"Structured ([green]{mode}[/green]): [cyan]{output_dir}[/cyan] ({doc_count} docs)"
158+
)
157159
console.print(f"Extractor: [green]{extractor}[/green]")
158160
if cats:
159161
console.print(f"Filtered to categories: {sorted(cats)}")

benchmarks/src/basic_memory_benchmarks/models.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,8 +177,12 @@ class ProviderDiagnosis(BaseModel):
177177
truly_missed: int = 0
178178
# Derived shares over answerable questions (0.0 when answerable == 0):
179179
qa_accuracy: float = 0.0 # correct / answerable
180-
retrieval_ceiling: float = 0.0 # (correct + retrieved_but_unused) / answerable — max QA if answerer were perfect
181-
answerer_gap: float = 0.0 # retrieved_but_unused / answerable — headroom left on the table by the answerer
180+
retrieval_ceiling: float = (
181+
0.0 # (correct + retrieved_but_unused) / answerable — max QA if answerer were perfect
182+
)
183+
answerer_gap: float = (
184+
0.0 # retrieved_but_unused / answerable — headroom left on the table by the answerer
185+
)
182186
retrieval_gap: float = 0.0 # truly_missed / answerable — headroom that needs better retrieval
183187
answerer_failure_share: float = 0.0 # retrieved_but_unused / (answerable failures) — of what we got wrong, how much was the answerer
184188
recall_field: str = "recall_at_10"

benchmarks/src/basic_memory_benchmarks/scoring/diagnose.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,8 +128,6 @@ def diagnose_run(
128128
seen.add(case.provider)
129129
providers.append(case.provider)
130130
return [
131-
diagnose_provider(
132-
qa_cases, retrieval_rows, provider=provider, recall_field=recall_field
133-
)
131+
diagnose_provider(qa_cases, retrieval_rows, provider=provider, recall_field=recall_field)
134132
for provider in providers
135133
]

benchmarks/test-int/test_compare_regression.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33

44
import pytest
55

6-
from basic_memory_benchmarks.reporting.compare import compare_provider_metric, load_retrieval_summary
6+
from basic_memory_benchmarks.reporting.compare import (
7+
compare_provider_metric,
8+
load_retrieval_summary,
9+
)
710

811

912
def test_compare_provider_metric(tmp_path: Path) -> None:

benchmarks/test-int/test_runner_orchestration.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,11 @@ def search(self, query: str, limit: int, run_config: RunConfig) -> list[SearchHi
2121
doc_id = self.mapping.get(query)
2222
if not doc_id:
2323
return []
24-
return [SearchHit(source_doc_id=doc_id, source_path=f"docs/{doc_id}.md", text="match", score=1.0)]
24+
return [
25+
SearchHit(
26+
source_doc_id=doc_id, source_path=f"docs/{doc_id}.md", text="match", score=1.0
27+
)
28+
]
2529

2630
def cleanup(self, run_config: RunConfig) -> None:
2731
_ = run_config

tests/mcp/test_project_context.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3199,9 +3199,7 @@ async def fake_index(context=None, force_refresh=False):
31993199
monkeypatch.setattr(project_context, "_ensure_workspace_project_index", fake_index)
32003200

32013201
# By external_id (the project_id routing path that failed in the field)
3202-
resolved = await resolve_workspace_project_identifier(
3203-
"22222222-2222-4222-8222-222222222222"
3204-
)
3202+
resolved = await resolve_workspace_project_identifier("22222222-2222-4222-8222-222222222222")
32053203
assert resolved.project.permalink == "manual"
32063204
assert calls == [False, True]
32073205

tests/mcp/test_tool_write_note.py

Lines changed: 24 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -19,29 +19,38 @@
1919

2020
def test_write_note_workspace_project_route_passthrough_without_workspace():
2121
"""Without workspace, the project string passes through unchanged."""
22-
assert _compose_workspace_project_route(
23-
workspace=None,
24-
project="my-project",
25-
project_id=None,
26-
) == "my-project"
22+
assert (
23+
_compose_workspace_project_route(
24+
workspace=None,
25+
project="my-project",
26+
project_id=None,
27+
)
28+
== "my-project"
29+
)
2730

2831

2932
def test_write_note_workspace_project_route_combines_workspace_and_project():
3033
"""workspace + project are joined as 'workspace/project'."""
31-
assert _compose_workspace_project_route(
32-
workspace="acme",
33-
project="docs",
34-
project_id=None,
35-
) == "acme/docs"
34+
assert (
35+
_compose_workspace_project_route(
36+
workspace="acme",
37+
project="docs",
38+
project_id=None,
39+
)
40+
== "acme/docs"
41+
)
3642

3743

3844
def test_write_note_workspace_project_route_passes_qualified_project_unchanged():
3945
"""A pre-qualified 'workspace/project' string passes through when workspace is None."""
40-
assert _compose_workspace_project_route(
41-
workspace=None,
42-
project="acme/docs",
43-
project_id=None,
44-
) == "acme/docs"
46+
assert (
47+
_compose_workspace_project_route(
48+
workspace=None,
49+
project="acme/docs",
50+
project_id=None,
51+
)
52+
== "acme/docs"
53+
)
4554

4655

4756
@pytest.mark.parametrize(

tests/repository/test_fastembed_provider.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -595,6 +595,7 @@ async def test_fastembed_provider_fails_fast_without_cache_dir(monkeypatch):
595595

596596
assert _SelfHealStubTextEmbedding.construct_count == 1
597597

598+
598599
@pytest.mark.asyncio
599600
async def test_factory_loads_native_model_once_across_repo_constructions(monkeypatch):
600601
"""The native ONNX model must load exactly once per process despite reuse (#872).

0 commit comments

Comments
 (0)