fixed error

allenai · Dec 17, 2024 · 2154ea8 · 2154ea8
1 parent 264ce05
commit 2154ea8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/convert_checkpoints_batch.py b/scripts/convert_checkpoints_batch.py
@@ -106,7 +106,7 @@ def convert_checkpoint(cps, load_dir="/data/input", sanity_check=False, weka_pre
                 except subprocess.CalledProcessError as e:
                     print(f"Error during checkpoint conversion: {checkpoint_path}")
                     error = {
-                        'error_code': e.return_code,
+                        'error_code': e.returncode,
                         'error_stderr': e.stderr
                     }
                     conversion_status = "error"