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(