Skip to content

Commit a63271c

Browse files
authored
Fix processing of models with '.' in their names
Fix issue where models with '.' in their names are not processed by scoring scripts
1 parent f46663e commit a63271c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

score_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def main() -> None:
109109

110110
model_names = []
111111
if args.all:
112-
model_names = [f.split('.')[0] for f in os.listdir('Models') if f.endswith('.json')]
112+
model_names = [f.removesuffix('.json') for f in os.listdir('Models') if f.endswith('.json')]
113113
elif args.models:
114114
model_names = args.models
115115

0 commit comments

Comments
 (0)