diff --git a/python_modules/libraries/dagster-fivetran/dagster_fivetran/translator.py b/python_modules/libraries/dagster-fivetran/dagster_fivetran/translator.py index b44d7390fafa9..554ae463fd768 100644 --- a/python_modules/libraries/dagster-fivetran/dagster_fivetran/translator.py +++ b/python_modules/libraries/dagster-fivetran/dagster_fivetran/translator.py @@ -20,7 +20,6 @@ class FivetranConnectorTableProps(NamedTuple): service: Optional[str] -@whitelist_for_serdes class FivetranConnectorSetupStateType(Enum): """Enum representing each setup state for a connector in Fivetran's ontology.""" @@ -38,7 +37,7 @@ class FivetranConnector: name: str service: str group_id: str - setup_state: FivetranConnectorSetupStateType + setup_state: str @property def url(self) -> str: @@ -50,7 +49,7 @@ def destination_id(self) -> str: @property def is_connected(self) -> bool: - return self.setup_state is FivetranConnectorSetupStateType.CONNECTED + return self.setup_state == FivetranConnectorSetupStateType.CONNECTED.value @classmethod def from_connector_details( @@ -62,7 +61,7 @@ def from_connector_details( name=connector_details["schema"], service=connector_details["service"], group_id=connector_details["group_id"], - setup_state=FivetranConnectorSetupStateType(connector_details["status"]["setup_state"]), + setup_state=connector_details["status"]["setup_state"], )