From 79785e478e9c34d32f85cb11705fbbb566bfc540 Mon Sep 17 00:00:00 2001
From: Susan Zhang <suchenzang@gmail.com>
Date: Sun, 12 Mar 2023 19:11:44 -0400
Subject: [PATCH] Setup.py update for multimodal work (#677)

* update setup.py from uriels PR #668

* add comment for installing extras
---
 setup.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 8cc013e7c..775204805 100644
--- a/setup.py
+++ b/setup.py
@@ -180,6 +180,7 @@ def do_setup(package_data):
             ]
         ),
         extras_require={
+            # install via: pip install -e ".[dev]"
             "dev": [
                 "flake8",
                 "black==22.3.0",
@@ -189,7 +190,17 @@ def do_setup(package_data):
                 "pyarrow",
                 "boto3",
                 "pandas",
-            ]
+            ],
+            # install via: pip install -e ".[multimodal]"
+            "multimodal": [
+                "albumentations",
+                "dalle_pytorch",
+                "einops",
+                "matplotlib",
+                "pytorchvideo==0.1.5",
+                "wandb",
+                "webdataset==0.1.103",
+            ],
         },
         package_data=package_data,
         ext_modules=extensions,