Revamp SUNDIALS and SuiteSparse installation script (#3719)

Replace wget with urllib for downloading and parallelize with concurrent.futures for efficiency

docs/source/user_guide/installation/install-from-source.rst

@@ -81,6 +81,24 @@ If you are running windows, you can simply skip this section and jump to :ref:`p
 This will download, compile and install the SuiteSparse and SUNDIALS libraries.
 Both libraries are installed in ``~/.local``.
 
+For users requiring more control over the installation process, the ``pybamm-requires`` session supports additional command-line arguments:
+
+- ``--install-dir``: Specify a custom installation directory for SUNDIALS and SuiteSparse.
+
+  Example:
+
+  .. code:: bash
+
+      nox -s pybamm-requires -- --install-dir [custom_directory_path]
+
+- ``--force``: Force the installation of SUNDIALS and SuiteSparse, even if they are already found in the specified directory.
+
+  Example:
+
+  .. code:: bash
+
+      nox -s pybamm-requires -- --force
+
 Manual install of build time requirements
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
@@ -98,6 +116,25 @@ PyBaMM ships with a Python script that automates points 2. and 3. You can run it
 
 	  python scripts/install_KLU_Sundials.py
 
+This script supports optional arguments for custom installations:
+
+- ``--install-dir``: Specify a custom installation directory for SUNDIALS and SuiteSparse.
+  By default, they are installed in ``~/.local``.
+
+  Example:
+
+  .. code:: bash
+
+      python scripts/install_KLU_Sundials.py --install-dir [custom_directory_path]
+
+- ``--force``: Force the installation of SUNDIALS and SuiteSparse, even if they are already found in the specified directory.
+
+  Example:
+
+  .. code:: bash
+
+      python scripts/install_KLU_Sundials.py --force
+
 .. _pybamm-install:
 
 Installing PyBaMM

noxfile.py

@@ -38,11 +38,11 @@ def set_environment_variables(env_dict, session):
 
 @nox.session(name="pybamm-requires")
 def run_pybamm_requires(session):
-    """Download, compile, and install the build-time requirements for Linux and macOS: the SuiteSparse and SUNDIALS libraries."""
+    """Download, compile, and install the build-time requirements for Linux and macOS. Supports --install-dir for custom installation paths and --force to force installation."""
     set_environment_variables(PYBAMM_ENV, session=session)
     if sys.platform != "win32":
-        session.install("wget", "cmake", silent=False)
-        session.run("python", "scripts/install_KLU_Sundials.py")
+        session.install("cmake", silent=False)
+        session.run("python", "scripts/install_KLU_Sundials.py", *session.posargs)
         if not os.path.exists("./pybind11"):
             session.run(
                 "git",