Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion m_flow/eval/report.py
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,7 @@ def _analyze_failures(
) -> tuple[List[FailureCase], Dict[str, int]]:
"""Analyze failure samples and bucket them."""
failures = []
buckets: Dict[str, int] = {k: 0 for k in FAILURE_BUCKETS}
buckets: Dict[str, int] = dict.fromkeys(FAILURE_BUCKETS, 0)

for case, result in zip(cases, results):
if not result.ok:
Expand Down
2 changes: 1 addition & 1 deletion m_flow/retrieval/episodic/bundle_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -338,7 +338,7 @@ async def search_in_collection(collection_name: str):

results = await asyncio.gather(*[search_in_collection(c) for c in cfg.collections])

node_distances = {c: r for c, r in zip(cfg.collections, results)}
node_distances = dict(zip(cfg.collections, results))
edge_distances = node_distances.get("RelationType_relationship_name", None)

return node_distances, edge_distances
Expand Down
2 changes: 1 addition & 1 deletion m_flow/retrieval/utils/fine_grained_triplet_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -252,7 +252,7 @@ async def search_in_collection(collection_name: str):
f"Vector collection retrieval completed: Retrieved distances from {sum(1 for res in results if res)} collections in {vector_collection_search_time:.2f}s"
)

node_distances = {collection: result for collection, result in zip(collections, results)}
node_distances = dict(zip(collections, results))

edge_distances = node_distances.get("RelationType_relationship_name", None)

Expand Down
2 changes: 1 addition & 1 deletion m_flow/retrieval/utils/procedural_bundle_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -286,7 +286,7 @@ async def search_col(col: str):
if all(not r for r in results):
return ([], []) if return_bundles else []

node_distances: Dict[str, list] = {c: r for c, r in zip(collections, results)}
node_distances: Dict[str, list] = dict(zip(collections, results))
edge_distances = node_distances.get("RelationType_relationship_name")

# Apply bonuses
Expand Down
2 changes: 1 addition & 1 deletion m_flow/tests/test_relational_db_migration.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@


def _nodes_dict(nodes):
return {n_id: data for (n_id, data) in nodes}
return dict(nodes)


def _normalize_name(name: str) -> str:
Expand Down
Loading