diff --git a/astronomer/providers/amazon/aws/operators/redshift_sql.py b/astronomer/providers/amazon/aws/operators/redshift_sql.py index 3d671bc38..e5a022cea 100644 --- a/astronomer/providers/amazon/aws/operators/redshift_sql.py +++ b/astronomer/providers/amazon/aws/operators/redshift_sql.py @@ -39,7 +39,7 @@ def __init__( ) -> None: self.redshift_conn_id = redshift_conn_id self.poll_interval = poll_interval - if self.__class__.__base__.__name__ == "RedshiftSQLOperator": + if self.__class__.__base__.__name__ == "RedshiftSQLOperator": # type: ignore[union-attr] # It's better to do str check of the parent class name because currently RedshiftSQLOperator # is deprecated and in future OSS RedshiftSQLOperator may be removed super().__init__(**kwargs) diff --git a/astronomer/providers/snowflake/operators/snowflake.py b/astronomer/providers/snowflake/operators/snowflake.py index dfab58de2..e13665188 100644 --- a/astronomer/providers/snowflake/operators/snowflake.py +++ b/astronomer/providers/snowflake/operators/snowflake.py @@ -130,7 +130,7 @@ def __init__( self.authenticator = authenticator self.session_parameters = session_parameters self.snowflake_conn_id = snowflake_conn_id - if self.__class__.__base__.__name__ != "SnowflakeOperator": + if self.__class__.__base__.__name__ != "SnowflakeOperator": # type: ignore[union-attr] # It's better to do str check of the parent class name because currently SnowflakeOperator # is deprecated and in future OSS SnowflakeOperator may be removed if any( @@ -319,7 +319,7 @@ def __init__( self.token_renewal_delta = token_renewal_delta self.bindings = bindings self.execute_async = False - if self.__class__.__base__.__name__ != "SnowflakeOperator": + if self.__class__.__base__.__name__ != "SnowflakeOperator": # type: ignore[union-attr] # It's better to do str check of the parent class name because currently SnowflakeOperator # is deprecated and in future OSS SnowflakeOperator may be removed if any(