Skip to content

Commit 6b2eb13

Browse files
committed
fix mypy
1 parent deccb76 commit 6b2eb13

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1915,17 +1915,25 @@ def create_declarative_stream(
19151915
else:
19161916
state_transformations = []
19171917

1918+
schema_loader: Union[
1919+
CompositeSchemaLoader,
1920+
DefaultSchemaLoader,
1921+
DynamicSchemaLoader,
1922+
InlineSchemaLoader,
1923+
JsonFileSchemaLoader,
1924+
]
19181925
if model.schema_loader and isinstance(model.schema_loader, list):
19191926
nested_schema_loaders = [
1920-
self._create_component_from_model(model=schema_loader, config=config)
1921-
for schema_loader in model.schema_loader
1927+
self._create_component_from_model(model=nested_schema_loader, config=config)
1928+
for nested_schema_loader in model.schema_loader
19221929
]
19231930
schema_loader = CompositeSchemaLoader(
19241931
schema_loaders=nested_schema_loaders, parameters={}
19251932
)
19261933
elif model.schema_loader:
19271934
schema_loader = self._create_component_from_model(
1928-
model=model.schema_loader, config=config
1935+
model=model.schema_loader, # type: ignore # If defined, schema_loader is guaranteed not to be a list and will be one of the existing base models
1936+
config=config,
19291937
)
19301938
else:
19311939
options = model.parameters or {}

0 commit comments

Comments
 (0)