python - 如何在 papermill 中参数化 Python 字典？

Question

我想传递一个变量中的所有参数。为了实现这一点，我使用以下方法在 Python 客户端中为 Papermill 传递字典:

params = {"aviral":"srivastava"}
pm.execute_notebook( 'path/to/input.ipynb', 'path/to/output.ipynb', parameters = params )

现在，我想从 bash 运行相应的命令。在 document它指出:

papermill local/input.ipynb s3://bkt/output.ipynb -p alpha 0.6 -p l1_ratio 0.1

但是没有给出如何传递字典的信息。

我尝试了很多排列和组合，例如:

papermill test_params.ipynb params_output.ipynb -p params {"aviral":"srivastava"}

papermill test_params.ipynb params_output.ipynb -p params '{"aviral":"srivastava"}'

我还阅读了以以下格式传递 dict : '{"aviral":"srivastava"}' ，然后 json.loads 会完成这项工作。但我没有转换输入的自由。它必须立即在脚本中使用。

只是一个最小的脚本来实际检查传递的变量的类型:

params = {} # I tag this with `parameter`
print(type(params))
print(params)

是的，我看到使用 papermill 运行时 params 变量发生了变化，但传递的值是一个完整的字符串。

Answer 1

由于在您的情况下，您无法直接修改模板笔记本来解析字符串输入，那么拥有一个您调用的 python 脚本然后执行 papermill 怎么样？这允许您将字符串解析为 python 字典，然后将其传递给 papermill 执行。您也可以以相同的方式传递笔记本名称。