Skip to content

Commit 1fc082b

Browse files
committed
chore: update black formatting
1 parent 342a0b5 commit 1fc082b

File tree

1 file changed

+20
-18
lines changed

1 file changed

+20
-18
lines changed

doc/doc_utils/jumpstart_doc_utils.py

Lines changed: 20 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -103,26 +103,26 @@ def get_model_task(id):
103103

104104

105105
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:
111111
return "HuggingFace"
112-
if 'catboost' in url:
112+
if "catboost" in url:
113113
return "Catboost"
114-
if 'gluon' in url:
114+
if "gluon" in url:
115115
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"
124124
else:
125-
return 'Source'
125+
return "Source"
126126

127127

128128
def create_jumpstart_model_table():
@@ -140,7 +140,7 @@ def create_jumpstart_model_table():
140140

141141
file_content = []
142142

143-
file_content.append('.. |external-link| raw:: html\n\n')
143+
file_content.append(".. |external-link| raw:: html\n\n")
144144
file_content.append(' <i class="fa fa-external-link"></i>\n\n')
145145

146146
file_content.append("==================================\n")
@@ -183,7 +183,9 @@ def create_jumpstart_model_table():
183183
file_content.append(" - {}\n".format(model["version"]))
184184
file_content.append(" - {}\n".format(model["min_version"]))
185185
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+
)
187189

188190
f = open("doc_utils/jumpstart.rst", "w")
189191
f.writelines(file_content)

0 commit comments

Comments
 (0)