From a45588d1598d5ae11effa859db01496cc2e87788 Mon Sep 17 00:00:00 2001
From: Lifepillar <lifepillar@users.noreply.github.com>
Date: Sun, 28 Jan 2024 08:58:35 +0000
Subject: [PATCH] Make Dask and Distributed optional dependencies (#1061)

---
 docs/text/quick_start.rst | 9 +++++++++
 setup.cfg                 | 6 ++++--
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/docs/text/quick_start.rst b/docs/text/quick_start.rst
index 890c46d4..05f5867e 100644
--- a/docs/text/quick_start.rst
+++ b/docs/text/quick_start.rst
@@ -13,6 +13,15 @@ As the compiled tsfresh package is hosted on the Python Package Index (PyPI) you
 
    pip install tsfresh
 
+If you need to work with large time series data that may not fit in memory, install tsfresh with
+`Dask <https://www.dask.org>`_:
+
+.. code:: shell
+
+   pip install tsfresh[dask]
+
+See also :ref:`large-data-label`.
+
 
 Dive in
 -------
diff --git a/setup.cfg b/setup.cfg
index e29e54e3..8eca9258 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -42,8 +42,6 @@ install_requires =
     patsy>=0.4.1
     scikit-learn>=0.22.0
     tqdm>=4.10.0
-    dask[dataframe]>=2.9.0
-    distributed>=2.11.0
     stumpy>=1.7.2
     cloudpickle
 
@@ -56,6 +54,9 @@ exclude =
     tests*
 
 [options.extras_require]
+dask =
+    dask[dataframe]>=2.9.0
+    distributed>=2.11.0
 matrixprofile =
     matrixprofile>=1.1.10,<2.0.0
 
@@ -67,6 +68,7 @@ docs =
 
 # Add here test requirements (semicolon/line-separated)
 testing =
+    %(dask)s
     pytest>=4.4.0
     pytest-cov>=2.6.1
     pytest-xdist>=1.26.1