Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
MinerU
Commits
c46c1534
Commit
c46c1534
authored
Jul 13, 2024
by
quyuan
Browse files
add ci
parent
d67af17f
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
13 deletions
+7
-13
tests/test_cli/test_bench.py
tests/test_cli/test_bench.py
+7
-13
No files found.
tests/test_cli/test_bench.py
View file @
c46c1534
...
...
@@ -11,9 +11,6 @@ from conf import conf
code_path
=
os
.
environ
.
get
(
'GITHUB_WORKSPACE'
)
pdf_dev_path
=
conf
.
conf
[
"pdf_dev_path"
]
pdf_res_path
=
conf
.
conf
[
"pdf_res_path"
]
last_simscore
=
0
last_editdistance
=
0
last_bleu
=
0
class
TestBench
():
"""
...
...
@@ -23,16 +20,13 @@ class TestBench():
"""
ci benchmark
"""
try
:
fr
=
open
(
os
.
path
.
join
(
pdf_dev_path
,
"result.json"
),
"r"
,
encoding
=
"utf-8"
)
lines
=
fr
.
readlines
()
last_line
=
lines
[
-
1
].
strip
()
last_score
=
json
.
loads
(
last_line
)
last_simscore
=
last_score
[
"average_sim_score"
]
last_editdistance
=
last_score
[
"average_edit_distance"
]
last_bleu
=
last_score
[
"average_bleu_score"
]
except
IOError
:
print
(
"result.json not exist"
)
fr
=
open
(
os
.
path
.
join
(
pdf_dev_path
,
"result.json"
),
"r"
,
encoding
=
"utf-8"
)
lines
=
fr
.
readlines
()
last_line
=
lines
[
-
1
].
strip
()
last_score
=
json
.
loads
(
last_line
)
last_simscore
=
last_score
[
"average_sim_score"
]
last_editdistance
=
last_score
[
"average_edit_distance"
]
last_bleu
=
last_score
[
"average_bleu_score"
]
os
.
system
(
f
"python tests/test_cli/lib/pre_clean.py --tool_name mineru --download_dir
{
pdf_dev_path
}
"
)
now_score
=
get_score
()
print
(
"now_score:"
,
now_score
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment