From 9e8342154f0e48ebf9382df5a2f9ecc3792d6085 Mon Sep 17 00:00:00 2001
From: Nicki Skafte Detlefsen <skaftenicki@gmail.com>
Date: Sun, 14 Apr 2024 16:31:03 +0200
Subject: [PATCH] Update `make docs` command (#2502)

---
 Makefile | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/Makefile b/Makefile
index 151d26a7e37..ea5bc674fb5 100644
--- a/Makefile
+++ b/Makefile
@@ -1,4 +1,4 @@
-.PHONY: test clean docs env data
+.PHONY: clean test pull-template docs env data
 
 export FREEZE_REQUIREMENTS=1
 # assume you have installed need packages
@@ -28,10 +28,14 @@ test: clean env data
 	cd tests && python -m pytest unittests -v --cov=torchmetrics
 	cd tests && python -m coverage report
 
-docs: clean
-	pip install -e . --quiet -r requirements/_docs.txt
+pull-template:
+	pip install -q awscli
+	aws s3 sync --no-sign-request s3://sphinx-packages/ dist/
+
+docs: clean pull-template
+	pip install -e . --quiet -r requirements/_docs.txt -f dist/
 	# apt-get install -y texlive-latex-extra dvipng texlive-pictures texlive-fonts-recommended cm-super
-	TOKENIZERS_PARALLELISM=false python -m sphinx -b html -W --keep-going docs/source docs/build
+	cd docs && make html --debug --jobs $(nproc) SPHINXOPTS="-W --keep-going"
 
 env:
 	pip install -e . -U -r requirements/_devel.txt