allow wandb to upload ckpts

allenai · Jul 5, 2024 · bf93d8d · bf93d8d
1 parent 8a2e393
commit bf93d8d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/allenact/algorithms/onpolicy_sync/runner.py b/allenact/algorithms/onpolicy_sync/runner.py
@@ -1205,6 +1205,7 @@ def update_keys_metric(
                     metrics=metric_dicts_list,
                     metric_means=callback_metric_means,
                     step=training_steps,
+                    checkpoint_file_name=checkpoint_file_name[0],
                     tasks_data=tasks_callback_data,
                     scalar_name_to_total_experiences_key=scalar_name_to_total_experiences_key,
                 )