8484
8585
8686class _IcebergExpressionVisitor (
87- _ExprVisitor ["BooleanExpression | UnboundTerm[Any] | Literal[Any] " ]
87+ _ExprVisitor ["BooleanExpression | UnboundTerm | Literal" ]
8888):
8989 """
9090 Visitor that converts Ray Data expressions to PyIceberg expressions.
@@ -99,11 +99,11 @@ class _IcebergExpressionVisitor(
9999 >>> # iceberg_expr can now be used with PyIceberg's filter APIs
100100 """
101101
102- def visit_column (self , expr : "ColumnExpr" ) -> "UnboundTerm[Any] " :
102+ def visit_column (self , expr : "ColumnExpr" ) -> "UnboundTerm" :
103103 """Convert a column reference to an Iceberg reference."""
104104 return Reference (expr .name )
105105
106- def visit_literal (self , expr : "LiteralExpr" ) -> "Literal[Any] " :
106+ def visit_literal (self , expr : "LiteralExpr" ) -> "Literal" :
107107 """Convert a literal value to an Iceberg literal."""
108108 return literal (expr .value )
109109
@@ -148,13 +148,11 @@ def visit_unary(self, expr: "UnaryExpr") -> "BooleanExpression":
148148
149149 def visit_alias (
150150 self , expr : "AliasExpr"
151- ) -> "BooleanExpression | UnboundTerm[Any] | Literal[Any] " :
151+ ) -> "BooleanExpression | UnboundTerm | Literal" :
152152 """Convert an aliased expression (just unwrap the alias)."""
153153 return self .visit (expr .expr )
154154
155- def visit_udf (
156- self , expr : "UDFExpr"
157- ) -> "BooleanExpression | UnboundTerm[Any] | Literal[Any]" :
155+ def visit_udf (self , expr : "UDFExpr" ) -> "BooleanExpression | UnboundTerm | Literal" :
158156 """UDF expressions cannot be converted to Iceberg expressions."""
159157 raise TypeError (
160158 "UDF expressions cannot be converted to Iceberg expressions. "
@@ -163,23 +161,23 @@ def visit_udf(
163161
164162 def visit_download (
165163 self , expr : "DownloadExpr"
166- ) -> "BooleanExpression | UnboundTerm[Any] | Literal[Any] " :
164+ ) -> "BooleanExpression | UnboundTerm | Literal" :
167165 """Download expressions cannot be converted to Iceberg expressions."""
168166 raise TypeError (
169167 "Download expressions cannot be converted to Iceberg expressions."
170168 )
171169
172170 def visit_star (
173171 self , expr : "StarExpr"
174- ) -> "BooleanExpression | UnboundTerm[Any] | Literal[Any] " :
172+ ) -> "BooleanExpression | UnboundTerm | Literal" :
175173 """Star expressions cannot be converted to Iceberg expressions."""
176174 raise TypeError (
177175 "Star expressions cannot be converted to Iceberg filter expressions."
178176 )
179177
180178 def visit_monotonically_increasing_id (
181179 self , expr : "MonotonicallyIncreasingIdExpr"
182- ) -> "BooleanExpression | UnboundTerm[Any] | Literal[Any] " :
180+ ) -> "BooleanExpression | UnboundTerm | Literal" :
183181 """Monotonically increasing ID expressions cannot be converted to Iceberg expressions."""
184182 raise TypeError (
185183 "monotonically_increasing_id expressions cannot be converted to Iceberg filter expressions."
@@ -490,7 +488,7 @@ def get_read_tasks(
490488 metadata = BlockMetadata (
491489 num_rows = sum (task .file .record_count for task in chunk_tasks )
492490 - position_delete_count ,
493- size_bytes = sum (task .length for task in chunk_tasks ),
491+ size_bytes = sum (task .file . file_size_in_bytes for task in chunk_tasks ),
494492 input_files = [task .file .file_path for task in chunk_tasks ],
495493 exec_stats = None ,
496494 )
0 commit comments