From 36833f949173cbf1685ef2fbe6865199a52e9b1b Mon Sep 17 00:00:00 2001
From: amogh7joshi <joshi.amoghn@gmail.com>
Date: Sun, 2 Oct 2022 10:32:19 -0400
Subject: [PATCH] Fix Helios errors and bump to v0.4.1

---
 MANIFEST.in      |  1 +
 agml/__init__.py |  2 +-
 setup.py         | 12 +-----------
 3 files changed, 3 insertions(+), 12 deletions(-)

diff --git a/MANIFEST.in b/MANIFEST.in
index ae93c7a6..f77ed185 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -1,4 +1,5 @@
 include README.md
 include agml/_assets/*.json
 include agml/_assets/shape_info.pickle
+include agml/_helios/helios_install.sh
 prune agml/_internal
\ No newline at end of file
diff --git a/agml/__init__.py b/agml/__init__.py
index 1eb74372..3023be68 100644
--- a/agml/__init__.py
+++ b/agml/__init__.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = '0.4.0'
+__version__ = '0.4.1'
 __all__ = ['data', 'backend', 'viz']
 
 
diff --git a/setup.py b/setup.py
index c4cf8967..4259f8d5 100644
--- a/setup.py
+++ b/setup.py
@@ -72,17 +72,7 @@
     packages=setuptools.find_packages(exclude = ('agml/_internal', )),
     include_package_data=True,
     classifiers=CLASSIFIERS,
-    install_requires=[
-        'numpy',
-        'matplotlib',
-        'tqdm',
-        'scikit-learn',
-        'opencv-python',
-        'torch',
-        'torchvision',
-        'tensorflow',
-        'pyyaml>=5.4.1'
-    ],
+    install_requires=install_requires,
     extras_require={
         'dev': dev_requires,
         'test': tests_require