Skip to content

Commit 9f9ac58

Browse files
author
maxime.c
committed
mypy again
1 parent 54f7b0f commit 9f9ac58

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2113,9 +2113,10 @@ def create_default_stream(
21132113
supports_file_transfer=hasattr(model, "file_uploader") and bool(model.file_uploader),
21142114
)
21152115

2116-
def _migrate_state(self, model: DeclarativeStreamModel, config: Config):
2116+
def _migrate_state(self, model: DeclarativeStreamModel, config: Config) -> None:
2117+
stream_name = model.name or ""
21172118
stream_state = self._connector_state_manager.get_stream_state(
2118-
stream_name=model.name, namespace=None
2119+
stream_name=stream_name, namespace=None
21192120
)
21202121
if model.state_migrations:
21212122
state_transformations = [
@@ -2126,7 +2127,7 @@ def _migrate_state(self, model: DeclarativeStreamModel, config: Config):
21262127
state_transformations = []
21272128
stream_state = self.apply_stream_state_migrations(state_transformations, stream_state)
21282129
self._connector_state_manager.update_state_for_stream(
2129-
stream_name=model.name, namespace=None, value=stream_state
2130+
stream_name=stream_name, namespace=None, value=stream_state
21302131
)
21312132

21322133
def _is_stop_condition_on_cursor(self, model: DeclarativeStreamModel) -> bool:

0 commit comments

Comments
 (0)