Fix pytype error.

mlcommons · Nov 29, 2024 · 0136e47 · 0136e47
1 parent 7ca7552
commit 0136e47
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/mlcroissant/mlcroissant/_src/datasets.py b/python/mlcroissant/mlcroissant/_src/datasets.py
@@ -223,7 +223,9 @@ def _filter_interesting_operations(self, filters: Filters | None) -> Operations:
         join_uuid = field.references.uuid
         graph = field.ctx.graph
         if join_uuid:
-            join_node = next(node for node in graph if node.uuid == join_uuid)
+            join_node = next(
+                node for node in graph if node.uuid == join_uuid
+            )  # pytype: disable=attribute-error
             unneeded_nodes = [
                 node
                 for node in graph