SageMakerのhyperparametersでlist型の引数を渡す方法

以下の issue の通り、 hyperparameters の各要素は一度 str に変換されます。

ですので、文字列にしたリストを list 型に戻す必要があります。このとき、json ライブラリを使うことで簡単に戻すことができます。

entrypoint.py

import json
import argparse

parser = argparse.ArgumentParser()
paresr.add_argument(
    '--listarg',
    type=lambda s: json.loads(s.replace("'", '"'))
)

変換するときに、シングルクォーテーションをダブルクォーテーションに置換しないとエラーが出てしまいます。

ちなみに、この方法で dict 型も引数として渡すことができます。