Make Dask and Distributed optional dependencies (#1061)

blue-yonder · Jan 28, 2024 · a45588d · a45588d
1 parent e2dfc6f
commit a45588d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git 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
@@ -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