Demo

action.yml

@@ -8,6 +8,10 @@ inputs:
       '--check --diff'"
     required: false
     default: "--check --diff"
+  jupyter:
+    description: "Install the required extra to run on Jupyter Notebooks."
+    required: false
+    default: false
   src:
     description: "Source to run Black. Default: '.'"
     required: false
@@ -38,6 +42,7 @@ runs:
         # TODO: Remove once https://github.com/actions/runner/issues/665 is fixed.
         INPUT_OPTIONS: ${{ inputs.options }}
         INPUT_SRC: ${{ inputs.src }}
+        INPUT_JUPYTER: ${{ inputs.jupyter }}
         INPUT_BLACK_ARGS: ${{ inputs.black_args }}
         INPUT_VERSION: ${{ inputs.version }}
         pythonioencoding: utf-8

action/main.py

@@ -10,11 +10,16 @@
 OPTIONS = os.getenv("INPUT_OPTIONS", default="")
 SRC = os.getenv("INPUT_SRC", default="")
 BLACK_ARGS = os.getenv("INPUT_BLACK_ARGS", default="")
+JUPYTER = os.getenv("INPUT_JUPYTER")
 VERSION = os.getenv("INPUT_VERSION", default="")
 
 run([sys.executable, "-m", "venv", str(ENV_PATH)], check=True)
 
-req = "black[colorama]"
+
+if JUPYTER:
+    req = "black[colorama,jupyter]"
+else:
+    req = "black[colorama]"
 if VERSION:
     req += f"=={VERSION}"
 pip_proc = run(