Skip to content

Commit 4bcc63a

Browse files
committed
Fix pytype error.
1 parent 7ca7552 commit 4bcc63a

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

python/mlcroissant/mlcroissant/_src/datasets.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,11 @@ def _filter_interesting_operations(self, filters: Filters | None) -> Operations:
223223
join_uuid = field.references.uuid
224224
graph = field.ctx.graph
225225
if join_uuid:
226-
join_node = next(node for node in graph if node.uuid == join_uuid)
226+
join_node = next(
227+
node
228+
for node in graph
229+
if isinstance(node, Field) and node.uuid == join_uuid
230+
)
227231
unneeded_nodes = [
228232
node
229233
for node in graph

0 commit comments

Comments
 (0)