Update generate_forecast_score_arrow.R

FLARE-forecast · Sep 20, 2024 · df7954f · df7954f
1 parent 559a4f3
commit df7954f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/generate_forecast_score_arrow.R b/R/generate_forecast_score_arrow.R
@@ -22,7 +22,7 @@ generate_forecast_score_arrow <- function(targets_df,
     vars <- arrow_env_vars()
     output_directory <- arrow::s3_bucket(bucket = bucket,
                                          endpoint_override =  endpoint)
-    unset_arrow_vars(vars)
+    on.exit(unset_arrow_vars(vars))
   }else{
     if(is.null(local_directory)){
       stop("scoring function needs local_directory if use_s3=FALSE")