From b5e5d10b60870e76ef9e907f3cac74e08f5aa14f Mon Sep 17 00:00:00 2001 From: Jackie Lam Date: Wed, 3 Apr 2024 11:38:09 -0700 Subject: [PATCH] remove extra argument passed to save_pkl_file function --- DocumentUnderstanding/VGT/object_detection/create_grid_input.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/DocumentUnderstanding/VGT/object_detection/create_grid_input.py b/DocumentUnderstanding/VGT/object_detection/create_grid_input.py index f04d85e..a655e9a 100644 --- a/DocumentUnderstanding/VGT/object_detection/create_grid_input.py +++ b/DocumentUnderstanding/VGT/object_detection/create_grid_input.py @@ -212,4 +212,4 @@ def select_tokenizer(tokenizer): for page in range(len(word_grid)): grid = create_grid_dict(tokenizer, word_grid[page]) - save_pkl_file(grid, args.output, f"page_{page}", page, args.model) + save_pkl_file(grid, args.output, f"page_{page}", args.model)