Skip to content

Commit

Permalink
Fix types & rst link
Browse files Browse the repository at this point in the history
  • Loading branch information
VolkerSchiewe committed May 7, 2024
1 parent 6fda2db commit 19f1f26
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 4 deletions.
8 changes: 5 additions & 3 deletions cosmos/operators/eks.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
from typing import Sequence
from __future__ import annotations

from typing import Sequence, Any

from airflow.exceptions import AirflowException
from airflow.providers.amazon.aws.hooks.eks import EksHook
Expand Down Expand Up @@ -31,7 +33,7 @@ def __init__(
namespace: str | None = DEFAULT_NAMESPACE,
aws_conn_id: str = DEFAULT_CONN_ID,
region: str | None = None,
**kwargs,
**kwargs: Any,
) -> None:
self.cluster_name = cluster_name
self.pod_name = pod_name
Expand All @@ -48,7 +50,7 @@ def __init__(
if self.config_file:
raise AirflowException("The config_file is not an allowed parameter for the EksPodOperator.")

def execute(self, context: Context):
def execute(self, context: Context) -> Any | None: # type: ignore
eks_hook = EksHook(
aws_conn_id=self.aws_conn_id,
region_name=self.region,
Expand Down
2 changes: 1 addition & 1 deletion docs/getting_started/execution-modes.rst
Original file line number Diff line number Diff line change
Expand Up @@ -168,7 +168,7 @@ EKS
----------

The ``eks`` approach is very similar to the ``kubernetes`` approach, but it is specifically designed to run on AWS EKS clusters.
It uses the [EKSPodOperator](https://airflow.apache.org/docs/apache-airflow-providers-amazon/2.2.0/operators/eks.html#perform-a-task-on-an-amazon-eks-cluster)
It uses the `EKSPodOperator <https://airflow.apache.org/docs/apache-airflow-providers-amazon/2.2.0/operators/eks.html#perform-a-task-on-an-amazon-eks-cluster>`_
to run the dbt commands. You need to provide the ``cluster_name`` in your operator_args to connect to the EKS cluster.


Expand Down

0 comments on commit 19f1f26

Please sign in to comment.