diff --git a/setup.py b/setup.py
index 2fe7b6cac14ade1728284fd6bac40d722eaeea16..3be2f287a2833676a8378bd8db04bde4451708ac 100644
--- a/setup.py
+++ b/setup.py
@@ -13,6 +13,8 @@ setup(
     include_package_data=True,
     install_requires=[
         'numpy',
+        'dask',
+        'pathos',
         'tqdm',
         'pandas',
         'scikit-image>=0.16.2',
@@ -23,6 +25,7 @@ setup(
         'imageio==2.8.0',
         'omero-py>=5.6.2',
         'zeroc-ice==3.6.5',
+        'tensorflow>=1.14,<=2.3'
         'baby@git+ssh://git@git.ecdf.ed.ac.uk/jpietsch/baby@dev',
         'logfile_parser@git+ssh://git@git.ecdf.ed.ac.uk/swain-lab/python-pipeline/logfile_parser.git',
         ],