Skip to content

Commit 11ba268

Browse files
authored
fix setup (#202)
1 parent 624e561 commit 11ba268

File tree

1 file changed

+26
-6
lines changed

1 file changed

+26
-6
lines changed

setup.py

Lines changed: 26 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -99,9 +99,6 @@ def get_libai_configs() -> List[str]:
9999
if __name__ == "__main__":
100100
print(f"Building wheel {package_name}-{version}")
101101

102-
with open("README.md", "r", encoding="utf-8") as f:
103-
readme = f.read()
104-
105102
with open("LICENSE", "r", encoding="utf-8") as f:
106103
license = f.read()
107104

@@ -111,10 +108,33 @@ def get_libai_configs() -> List[str]:
111108
name=package_name,
112109
version=version,
113110
description="Toolkit for Pretraining Models with OneFlow",
114-
long_description_content_type="text/markdown",
115-
long_description=readme,
116111
license=license,
117-
install_requires=requirements,
112+
install_requires=[
113+
"boto3",
114+
"botocore",
115+
"cloudpickle",
116+
"flowvision==0.1.0",
117+
"wget",
118+
"hydra-core",
119+
"nltk",
120+
"numpy",
121+
"omegaconf",
122+
"Pygments",
123+
"PyYAML",
124+
"jieba",
125+
"regex",
126+
"requests",
127+
"scipy",
128+
"sentencepiece>=0.1",
129+
"tabulate",
130+
"termcolor",
131+
"tqdm",
132+
"pybind11",
133+
"portalocker",
134+
"flake8==3.8.1 ",
135+
"isort==5.10.1",
136+
"black==21.4b ",
137+
],
118138
packages=find_packages(),
119139
package_data={"libai.config": get_libai_configs()},
120140
ext_modules=extensions,

0 commit comments

Comments
 (0)