From bbbafe54241f8eb66d65a06023d483b10249f862 Mon Sep 17 00:00:00 2001
From: Nguyen Huy Hoang <24520554@gm.uit.edu.vn>
Date: Thu, 26 Mar 2026 22:11:59 +0700
Subject: [PATCH 1/2] feat: extend parameter file loader to parse toml in
 addition to yaml

Signed-off-by: Nguyen Huy Hoang <181364121+huyhoang171106@users.noreply.github.com>
---
 papermill/iorw.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/papermill/iorw.py b/papermill/iorw.py
index 14a0122c..a737bcf7 100644
--- a/papermill/iorw.py
+++ b/papermill/iorw.py
@@ -5,6 +5,14 @@
 import warnings
 from contextlib import contextmanager
 
+try:
+    import tomllib
+except ModuleNotFoundError:
+    try:
+        import tomli as tomllib
+    except ModuleNotFoundError:
+        tomllib = None
+
 import entrypoints
 import nbformat
 import requests

From 46dba0c867948929cfc4624a6bc44528d1995a9f Mon Sep 17 00:00:00 2001
From: Nguyen Huy Hoang <24520554@gm.uit.edu.vn>
Date: Thu, 26 Mar 2026 22:12:01 +0700
Subject: [PATCH 2/2] feat: extend parameter file loader to parse toml in
 addition to yaml

Signed-off-by: Nguyen Huy Hoang <181364121+huyhoang171106@users.noreply.github.com>
---
 papermill/cli.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/papermill/cli.py b/papermill/cli.py
index 0c1b54d0..589de063 100755
--- a/papermill/cli.py
+++ b/papermill/cli.py
@@ -42,7 +42,7 @@ def print_papermill_version(ctx, param, value):
 )
 @click.option('--parameters', '-p', nargs=2, multiple=True, help='Parameters to pass to the parameters cell.')
 @click.option('--parameters_raw', '-r', nargs=2, multiple=True, help='Parameters to be read as raw string.')
-@click.option('--parameters_file', '-f', multiple=True, help='Path to YAML file containing parameters.')
+@click.option('--parameters_file', '-f', multiple=True, help='Path to YAML or TOML file containing parameters.')
 @click.option('--parameters_yaml', '-y', multiple=True, help='YAML string to be used as parameters.')
 @click.option('--parameters_base64', '-b', multiple=True, help='Base64 encoded YAML string as parameters.')
 @click.option(