Skip to content

Commit 14b5b35

Browse files
committed
fix: filepath
1 parent 1d838d4 commit 14b5b35

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

binaryai_bindiffmatch/metricsutils.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,12 +56,13 @@ def build_testcase_cross_optimization_pairs_on_library(
5656

5757
def get_stripped_filenames(datadir: str, library: str) -> list[str]:
5858
# assert library in libraries
59-
stripped_binary_relpath = get_stripped_binary_relpath(
59+
labeled_doc_relpath = get_labeled_doc_relpath(
6060
library, versions[library][-1], optimazations[library][-1], None
6161
)
6262
result = []
63-
for filename in os.listdir(os.path.join(datadir, stripped_binary_relpath)):
64-
assert filename.endswith(".strip")
63+
for filename in os.listdir(os.path.join(datadir, labeled_doc_relpath)):
64+
assert filename.endswith(".strip.json")
65+
filename = filename.removesuffix(".json")
6566
result.append(filename)
6667
return result
6768

0 commit comments

Comments
 (0)