fix: pass args.data_folder to corpus_iterator

swerik-project · Jan 14, 2025 · 204dd6c · 204dd6c
1 parent c0569bd
commit 204dd6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyriksdagen/args.py b/pyriksdagen/args.py
@@ -107,7 +107,7 @@ def common_args(args):
         for py in args.parliament_year:
             args.specific_files.extend(glob(f"{args.data_folder}/{py}/*.xml"))
     else:
-        args.specific_files = sorted(list(corpus_iterator(args.doctype, start=args.start, end=args.end)))
+        args.specific_files = sorted(list(corpus_iterator(args.doctype, corpus_root=args.data_folder, start=args.start, end=args.end)))
     if args.specific_files is not None and len(list(args.specific_files)) != 0:
         args.specific_files = sorted(list(args.specific_files))
     return rename_file_list(args)