File tree Expand file tree Collapse file tree 1 file changed +26
-6
lines changed Expand file tree Collapse file tree 1 file changed +26
-6
lines changed Original file line number Diff line number Diff line change @@ -99,9 +99,6 @@ def get_libai_configs() -> List[str]:
99
99
if __name__ == "__main__" :
100
100
print (f"Building wheel { package_name } -{ version } " )
101
101
102
- with open ("README.md" , "r" , encoding = "utf-8" ) as f :
103
- readme = f .read ()
104
-
105
102
with open ("LICENSE" , "r" , encoding = "utf-8" ) as f :
106
103
license = f .read ()
107
104
@@ -111,10 +108,33 @@ def get_libai_configs() -> List[str]:
111
108
name = package_name ,
112
109
version = version ,
113
110
description = "Toolkit for Pretraining Models with OneFlow" ,
114
- long_description_content_type = "text/markdown" ,
115
- long_description = readme ,
116
111
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
+ ],
118
138
packages = find_packages (),
119
139
package_data = {"libai.config" : get_libai_configs ()},
120
140
ext_modules = extensions ,
You can’t perform that action at this time.
0 commit comments