Skip to content

Commit

Permalink
fix: typing
Browse files Browse the repository at this point in the history
  • Loading branch information
JasperHG90 committed Oct 21, 2024
1 parent 35a5e68 commit 594ec34
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions packages/dagster_pyiceberg/src/dagster_pyiceberg/handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@
from pyiceberg.exceptions import CommitFailedException
from pyiceberg.partitioning import PartitionSpec
from pyiceberg.schema import Schema
from pyiceberg.table.update.schema import UpdateSchema
from pyiceberg.table.update.spec import UpdateSpec
from tenacity import RetryError, Retrying, stop_after_attempt, wait_random

Expand Down Expand Up @@ -204,15 +205,15 @@ def __init__(
self.schema_differ = schema_differ

@staticmethod
def _delete_column(update: table.UpdateSchema, column: str):
def _delete_column(update: UpdateSchema, column: str):
try:
update.delete_column(column)
except ValueError:
# Already deleted by another operation
pass

@staticmethod
def _merge_schemas(update: table.UpdateSchema, new_table_schema: pa.Schema):
def _merge_schemas(update: UpdateSchema, new_table_schema: pa.Schema):
try:
update.union_by_name(new_table_schema)
except ValueError:
Expand Down

0 comments on commit 594ec34

Please sign in to comment.