diff --git a/python_modules/libraries/dagster-fivetran/dagster_fivetran/__init__.py b/python_modules/libraries/dagster-fivetran/dagster_fivetran/__init__.py index 265c5056c1fad..14b96afe5b1ea 100644 --- a/python_modules/libraries/dagster-fivetran/dagster_fivetran/__init__.py +++ b/python_modules/libraries/dagster-fivetran/dagster_fivetran/__init__.py @@ -3,6 +3,7 @@ from dagster_fivetran.asset_decorator import fivetran_assets as fivetran_assets from dagster_fivetran.asset_defs import ( build_fivetran_assets as build_fivetran_assets, + build_fivetran_assets_definitions as build_fivetran_assets_definitions, load_assets_from_fivetran_instance as load_assets_from_fivetran_instance, ) from dagster_fivetran.ops import ( diff --git a/python_modules/libraries/dagster-fivetran/dagster_fivetran/asset_decorator.py b/python_modules/libraries/dagster-fivetran/dagster_fivetran/asset_decorator.py index 0a23f665553a6..86c8929421dac 100644 --- a/python_modules/libraries/dagster-fivetran/dagster_fivetran/asset_decorator.py +++ b/python_modules/libraries/dagster-fivetran/dagster_fivetran/asset_decorator.py @@ -1,4 +1,4 @@ -from typing import Any, Callable, Optional, Type +from typing import Any, Callable, Optional from dagster import AssetsDefinition, multi_asset from dagster._annotations import experimental @@ -107,7 +107,8 @@ def fivetran_connector_assets(context: dg.AssetExecutionContext, fivetran: Fivet specs=[ spec for spec in workspace.load_asset_specs( - dagster_fivetran_translator=dagster_fivetran_translator or DagsterFivetranTranslator() + dagster_fivetran_translator=dagster_fivetran_translator + or DagsterFivetranTranslator() ) if FivetranMetadataSet.extract(spec.metadata).connector_id == connector_id ],