added data downloading

setup.py

@@ -1,7 +1,9 @@
 import setuptools
 import os
 from io import open
+from setuptools.command.install import install
 from setuptools import find_packages
+from download_data import download_data
 
 src_dir = os.path.abspath(os.path.dirname(__file__))
 
@@ -15,24 +17,35 @@ if os.path.isfile(requirements_path):
     with open(requirements_path) as f:
         requirements = f.read().splitlines()
 
+class PostInstallCommand(install):
+    """Post-installation for installation mode."""
+    def run(self):
+        print("RUNNING POST INSTALL 1")
+        install.run(self)
+        print("RUNNING POST INSTALL")
+        download_data()
+
 
 setuptools.setup(
-    name="fig-benchmark",
+    name="genies-benchmark",
     version="0.0.1",
     author="Joshua Clymer, Garrett Baker, Rohan Subramani, and Sam Wang",
     author_email="joshuamclymer@gmail.com",
     description="The fig benchmark repository contains datasets and tooling for evaluating the generalization of preferrence models.",
     long_description=long_description,
     long_description_content_type="text/markdown",
-    url="https://github.com/Joshuaclymer/FIG-benchmark",
+    url="https://github.com/Joshuaclymer/GENIES",
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: MIT License",
         "Operating System :: OS Independent",
     ],
+    cmdclass={
+        'install': PostInstallCommand,
+    },
     python_requires='>=3.10',
     install_requires=requirements,
     packages=find_packages(where='src'),  # Specify 'src' as the root
     package_dir={'': 'src'},
-    package_data={'fig-benchmark': ['LICENCE', 'requirements.txt']},
+    package_data={'genies-benchmark': ['LICENCE', 'requirements.txt']},
 )