airflow.providers.amazon.aws.hooks.sagemaker_unified_studio¶

此模組包含 Amazon SageMaker Unified Studio Notebook 的 Hook。

類¶

SageMakerNotebookHook

與 Sagemaker Unified Studio 工作流互動。

模組內容¶

class airflow.providers.amazon.aws.hooks.sagemaker_unified_studio.SageMakerNotebookHook(execution_name, input_config=None, output_config=None, compute=None, termination_condition=None, tags=None, waiter_delay=10, waiter_max_attempts=1440, *args, **kwargs)[源]¶

基類: airflow.hooks.base.BaseHook

與 Sagemaker Unified Studio 工作流互動。

此 Hook 封裝了 Sagemaker 工作流 Notebook 執行 API。

示例

from airflow.providers.amazon.aws.hooks.sagemaker_unified_studio import SageMakerNotebookHook

notebook_hook = SageMakerNotebookHook(
    input_config={"input_path": "path/to/notebook.ipynb", "input_params": {"param1": "value1"}},
    output_config={"output_uri": "folder/output/location/prefix", "output_formats": "NOTEBOOK"},
    execution_name="notebook_execution",
    waiter_delay=10,
    waiter_max_attempts=1440,
)

引數:

execution_name (str) – 要執行的 notebook 作業名稱，與 task_id 相同。
input_config (dict | None) – 輸入檔案的配置。示例：{'input_path': 'folder/input/notebook.ipynb', 'input_params': {'param1': 'value1'}}
output_config (dict | None) – 輸出格式的配置。應包含一個 output_formats 引數來指定輸出格式。示例：{'output_formats': ['NOTEBOOK']}
compute (dict | None) – 用於 notebook 執行的計算配置。如果在遠端計算上執行，這是必需屬性。示例：{ "instance_type": "ml.m5.large", "volume_size_in_gb": 30, "volume_kms_key_id": "", "image_uri": "string", "container_entrypoint": [ "string" ]}
termination_condition (dict | None) – 用於匹配以終止遠端執行的條件。示例：{ "MaxRuntimeInSeconds": 3600 }
tags (dict | None) – 與遠端執行執行關聯的標籤。示例：{ "md_analytics": "logs" }
waiter_delay (int) – 檢查任務執行狀態的時間間隔（秒）。
waiter_max_attempts (int) – 在返回 FAILED 之前等待的嘗試次數。

execution_name[源]¶

input_config[源]¶

output_config[源]¶

compute = None[源]¶

termination_condition[源]¶

tags[源]¶

waiter_delay = 10[源]¶

waiter_max_attempts = 1440[源]¶

start_notebook_execution()[源]¶

wait_for_execution_completion(execution_id, context)[源]¶