Skip to content

Commit a32bb06

Browse files
committed
fix: type hints
1 parent 763042f commit a32bb06

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

pyiceberg/table/upsert_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535

3636
def create_match_filter(df: pyarrow_table, join_cols: list[str]) -> BooleanExpression:
3737
unique_keys = df.select(join_cols).group_by(join_cols).aggregate([])
38-
filters = []
38+
filters: list[BooleanExpression] = []
3939

4040
if len(join_cols) == 1:
4141
column = join_cols[0]

tests/table/test_upsert.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
# specific language governing permissions and limitations
1616
# under the License.
1717
from pathlib import PosixPath
18+
from typing import Any
1819

1920
import pyarrow as pa
2021
import pytest
@@ -494,7 +495,7 @@ def test_create_match_filter_single_condition() -> None:
494495
),
495496
],
496497
)
497-
def test_create_match_filter_with_nulls(data: list[dict], expected: BooleanExpression) -> None:
498+
def test_create_match_filter_with_nulls(data: list[dict[str, Any]], expected: BooleanExpression) -> None:
498499
schema = pa.schema([pa.field("x", pa.float64()), pa.field("y", pa.float64())])
499500
table = pa.Table.from_pylist(data, schema=schema)
500501
join_cols = sorted({col for record in data for col in record})

0 commit comments

Comments
 (0)