Skip to content

Commit 5b63bd4

Browse files
committed
Fix data preparation script
1 parent be71986 commit 5b63bd4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/cpp/precommit/prepare_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def prepare_model(
3030
public_scope = json.load(f)
3131

3232
for model in public_scope:
33-
if model["name"].endswith(".xml"):
33+
if model["name"].endswith(".xml") or model["name"].endswith(".onnx"):
3434
continue
3535
model = eval(model["type"]).create_model(model["name"], download_dir=data_dir)
3636

0 commit comments

Comments
 (0)