ノートブックジョブで別のノートブックを呼び出す

あるノートブックジョブが別のノートブックを呼び出すようなパイプラインを設定できます。以下では、ノートブックが他の 2 つのノートブックを呼び出す、ノートブックジョブステップを含むパイプラインの例を設定します。入力ノートブックには次の行が含まれます。

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

次のスニペットに示されるとおり、additional_dependencies を使用してこれらのノートブックを NotebookJobStep インスタンスに渡します。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,
)