mirror of
https://github.com/wassname/GENIES.git
synced 2026-06-27 16:10:25 +08:00
Joshuaclymer
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
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__))
|
|
|
|
with open("README.md", "r") as fh:
|
|
long_description = fh.read()
|
|
|
|
# Build requirements
|
|
requirements_path = f"{src_dir}/requirements.txt"
|
|
requirements = []
|
|
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="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/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={'genies-benchmark': ['LICENCE', 'requirements.txt']},
|
|
) |