Skip to content

Commit 13804d4

Browse files
authored
chore(weave): ruff: enable C419 (#2908)
1 parent 4033e92 commit 13804d4

File tree

4 files changed

+12
-11
lines changed

4 files changed

+12
-11
lines changed

examples/weaveflow/get_started.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@
157157
"@weave.op()\n",
158158
"def aggregate_scores(score_dicts) -> float:\n",
159159
" return sum(\n",
160-
" [score_dict[\"brevity\"] * score_dict[\"relevance\"] for score_dict in score_dicts]\n",
160+
" score_dict[\"brevity\"] * score_dict[\"relevance\"] for score_dict in score_dicts\n",
161161
" ) / len(score_dicts)\n",
162162
"\n",
163163
"\n",

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -169,6 +169,7 @@ select = [
169169
"UP", # https://docs.astral.sh/ruff/rules/#pyupgrade-up
170170
"C409", # https://docs.astral.sh/ruff/rules/unnecessary-literal-within-tuple-call/
171171
"C400", # https://docs.astral.sh/ruff/rules/unnecessary-generator-list/
172+
"C419", # https://docs.astral.sh/ruff/rules/unnecessary-comprehension-in-call/
172173
]
173174
ignore = [
174175
# we use Google style

tests/trace/test_client_trace.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -334,7 +334,7 @@ def liner(m: Number, b, x) -> Number:
334334
run_calls += num_calls * 3
335335
root_calls += num_calls
336336

337-
total_calls = sum([op_call.num_calls for op_call in result.values()])
337+
total_calls = sum(op_call.num_calls for op_call in result.values())
338338

339339
return OpCallSpec(
340340
call_summaries=result,
@@ -409,7 +409,7 @@ def test_trace_call_query_filter_op_version_refs(client):
409409

410410

411411
def has_any(list_a: list[str], list_b: list[str]) -> bool:
412-
return any([a in list_b for a in list_a])
412+
return any(a in list_b for a in list_a)
413413

414414

415415
def unique_vals(list_a: list[str]) -> list[str]:
@@ -425,7 +425,7 @@ def get_all_calls_asserting_finished(
425425
)
426426
)
427427
assert len(res.calls) == call_spec.total_calls
428-
assert all([call.ended_at for call in res.calls])
428+
assert all(call.ended_at for call in res.calls)
429429
return res
430430

431431

tests/trace/test_objs_query.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def test_objs_query_filter_object_ids(client: WeaveClient):
3030
)
3131
)
3232
assert len(res.objs) == 20
33-
assert all([obj.object_id in ["obj_0", "obj_1"] for obj in res.objs])
33+
assert all(obj.object_id in ["obj_0", "obj_1"] for obj in res.objs)
3434

3535

3636
def test_objs_query_filter_is_op(client: WeaveClient):
@@ -60,8 +60,8 @@ def test_objs_query_filter_latest_only(client: WeaveClient):
6060
)
6161
)
6262
assert len(res.objs) == 10
63-
assert all([obj.is_latest for obj in res.objs])
64-
assert all([obj.val["j"] == 9 for obj in res.objs])
63+
assert all(obj.is_latest for obj in res.objs)
64+
assert all(obj.val["j"] == 9 for obj in res.objs)
6565

6666

6767
def test_objs_query_filter_limit_offset_sort_by_created_at(client: WeaveClient):
@@ -77,7 +77,7 @@ def test_objs_query_filter_limit_offset_sort_by_created_at(client: WeaveClient):
7777
)
7878
)
7979
assert len(res.objs) == 3
80-
assert all([obj.is_latest for obj in res.objs])
80+
assert all(obj.is_latest for obj in res.objs)
8181
assert res.objs[0].val["j"] == 9
8282
assert res.objs[0].val["i"] == 4
8383
assert res.objs[1].val["j"] == 9
@@ -95,7 +95,7 @@ def test_objs_query_filter_limit_offset_sort_by_created_at(client: WeaveClient):
9595
)
9696
)
9797
assert len(res.objs) == 3
98-
assert all([obj.is_latest for obj in res.objs])
98+
assert all(obj.is_latest for obj in res.objs)
9999
assert res.objs[0].val["j"] == 9
100100
assert res.objs[0].val["i"] == 5
101101
assert res.objs[1].val["j"] == 9
@@ -117,7 +117,7 @@ def test_objs_query_filter_limit_offset_sort_by_object_id(client: WeaveClient):
117117
)
118118
)
119119
assert len(res.objs) == 3
120-
assert all([obj.is_latest for obj in res.objs])
120+
assert all(obj.is_latest for obj in res.objs)
121121
assert res.objs[0].val["j"] == 9
122122
assert res.objs[0].val["i"] == 4
123123
assert res.objs[1].val["j"] == 9
@@ -135,7 +135,7 @@ def test_objs_query_filter_limit_offset_sort_by_object_id(client: WeaveClient):
135135
)
136136
)
137137
assert len(res.objs) == 3
138-
assert all([obj.is_latest for obj in res.objs])
138+
assert all(obj.is_latest for obj in res.objs)
139139
assert res.objs[0].val["j"] == 9
140140
assert res.objs[0].val["i"] == 5
141141
assert res.objs[1].val["j"] == 9

0 commit comments

Comments
 (0)