Skip to content

Commit c3a7d19

Browse files
committed
fix(types): filter None values in alternatives list comprehension
1 parent 00031cc commit c3a7d19

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

redisvl/extensions/router/semantic.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -333,7 +333,9 @@ def _apply_cost_optimization(self, matches: List[RouteMatch]) -> List[RouteMatch
333333
# Update alternatives with original matches
334334
for i, (match, _) in enumerate(ranked):
335335
match.alternatives = [
336-
(m.name, m.distance) for j, (m, _) in enumerate(ranked) if j != i
336+
(m.name, m.distance)
337+
for j, (m, _) in enumerate(ranked)
338+
if j != i and m.name is not None and m.distance is not None
337339
]
338340

339341
return [m for m, _ in ranked]
@@ -1389,7 +1391,9 @@ def _apply_cost_optimization(self, matches: List[RouteMatch]) -> List[RouteMatch
13891391
# Update alternatives with original matches
13901392
for i, (match, _) in enumerate(ranked):
13911393
match.alternatives = [
1392-
(m.name, m.distance) for j, (m, _) in enumerate(ranked) if j != i
1394+
(m.name, m.distance)
1395+
for j, (m, _) in enumerate(ranked)
1396+
if j != i and m.name is not None and m.distance is not None
13931397
]
13941398

13951399
return [m for m, _ in ranked]

0 commit comments

Comments
 (0)