View a markdown version of this page

在您的筆記本任務中調用另一個筆記本 - Amazon SageMaker AI

本文為英文版的機器翻譯版本，如內容有任何歧義或不一致之處，概以英文版為準。

在您的筆記本任務中調用另一個筆記本

您可以設定其中一個筆記本任務呼叫另一個筆記本的管道。以下設定具有筆記本任務步驟的管道範例，其中筆記本會呼叫其他兩個筆記本。輸入筆記本包含下列幾行：

%run 'subfolder/notebook_to_call_in_subfolder.ipynb'
%run 'notebook_to_call.ipynb'

使用 NotebookJobStep 將這些筆記本傳遞至您的 additional_dependencies 執行個體，如下列程式碼片段所示。請注意，additional_dependencies 中為筆記本提供的路徑是從根位置提供的。如需 SageMaker AI 如何將您的相依檔案和資料夾上傳至 Amazon S3，以便您可以正確提供相依性路徑的相關資訊，請參閱 NotebookJobStep 中 additional_dependencies 的描述。

input_notebook = "inputs/input_notebook.ipynb"
simple_notebook_path = "inputs/notebook_to_call.ipynb"
folder_with_sub_notebook = "inputs/subfolder"

notebook_job_step = NotebookJobStep(
    image_uri=image-uri,
    kernel_name=kernel-name,
    role=role-name,
    input_notebook=input_notebook,
    additional_dependencies=[simple_notebook_path, folder_with_sub_notebook],
    tags=tags,
)