forked from ray-project/ray_shuffling_data_loader
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
23 lines (22 loc) · 734 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from setuptools import find_packages, setup
setup(
name="ray_shuffling_data_loader",
version="0.1.0",
author="Ray Team",
author_email="ray-dev@googlegroups.com",
description="A Ray-based data loader with pipelined per-epoch shuffling.",
long_description=(
"A Ray-based data loader with per-epoch shuffling and configurable "
"pipelining, for shuffling and loading training data for distributed "
"training of machine learning models."),
url="https://github.com/ray-project/ray_shuffling_data_loader",
install_requires=[
"ray",
"numpy",
"pandas",
"smart-open",
"torch",
],
packages=find_packages(),
python_requires=">=3.6",
)