From d098f3f6a518a00550d0e12c7dba8f842904c258 Mon Sep 17 00:00:00 2001
From: Nicolas Viennot <Nicolas.Viennot@twosigma.com>
Date: Fri, 14 May 2021 23:13:11 +0000
Subject: [PATCH] Prepare the image directory during checkpoint

---
 src/cli/checkpoint.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/cli/checkpoint.rs b/src/cli/checkpoint.rs
index e346b3f..f532cb0 100644
--- a/src/cli/checkpoint.rs
+++ b/src/cli/checkpoint.rs
@@ -141,6 +141,7 @@ pub fn do_checkpoint(opts: Checkpoint) -> Result<Stats> {
         num_shards, passphrase_file.is_some(), cpu_budget.into());
 
     let store = image_url.store();
+    store.prepare(true)?;
     let shard_upload_cmds = shard::upload_cmds(
         &img_manifest, passphrase_file.as_ref(), &*store)?;