File tree Expand file tree Collapse file tree 1 file changed +20
-18
lines changed Expand file tree Collapse file tree 1 file changed +20
-18
lines changed Original file line number Diff line number Diff line change @@ -103,26 +103,26 @@ def get_model_task(id):
103
103
104
104
105
105
def get_model_source (url ):
106
- if ' tfhub' in url :
107
- return ' Tensorflow Hub'
108
- if ' pytorch' in url :
109
- return ' Pytorch Hub'
110
- if ' huggingface' in url :
106
+ if " tfhub" in url :
107
+ return " Tensorflow Hub"
108
+ if " pytorch" in url :
109
+ return " Pytorch Hub"
110
+ if " huggingface" in url :
111
111
return "HuggingFace"
112
- if ' catboost' in url :
112
+ if " catboost" in url :
113
113
return "Catboost"
114
- if ' gluon' in url :
114
+ if " gluon" in url :
115
115
return "GluonCV"
116
- if ' catboost' in url :
117
- return ' Catboost'
118
- if ' lightgbm' in url :
119
- return ' LightGBM'
120
- if ' xgboost' in url :
121
- return ' XGBoost'
122
- if ' scikit' in url :
123
- return ' ScikitLearn'
116
+ if " catboost" in url :
117
+ return " Catboost"
118
+ if " lightgbm" in url :
119
+ return " LightGBM"
120
+ if " xgboost" in url :
121
+ return " XGBoost"
122
+ if " scikit" in url :
123
+ return " ScikitLearn"
124
124
else :
125
- return ' Source'
125
+ return " Source"
126
126
127
127
128
128
def create_jumpstart_model_table ():
@@ -140,7 +140,7 @@ def create_jumpstart_model_table():
140
140
141
141
file_content = []
142
142
143
- file_content .append (' .. |external-link| raw:: html\n \n ' )
143
+ file_content .append (" .. |external-link| raw:: html\n \n " )
144
144
file_content .append (' <i class="fa fa-external-link"></i>\n \n ' )
145
145
146
146
file_content .append ("==================================\n " )
@@ -183,7 +183,9 @@ def create_jumpstart_model_table():
183
183
file_content .append (" - {}\n " .format (model ["version" ]))
184
184
file_content .append (" - {}\n " .format (model ["min_version" ]))
185
185
file_content .append (" - {}\n " .format (model_task ))
186
- file_content .append (" - `{} <{}>`__ |external-link|\n " .format (model_source , model_spec ["url" ]))
186
+ file_content .append (
187
+ " - `{} <{}>`__ |external-link|\n " .format (model_source , model_spec ["url" ])
188
+ )
187
189
188
190
f = open ("doc_utils/jumpstart.rst" , "w" )
189
191
f .writelines (file_content )
You can’t perform that action at this time.
0 commit comments