Added cleanup for input image after stub

Signed-off-by: Anmol Virdi <[email protected]>
rapidfort · Jul 1, 2024 · c04b8e5 · c04b8e5
1 parent f49004b
commit c04b8e5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/community_images/common/orchestrator/stub_generator.py b/community_images/common/orchestrator/stub_generator.py
@@ -80,6 +80,7 @@ def generate_stub_for_tag_mappings(self, tag_mappings):
 
             # Remove original input image to clean up space
             logging.info(f"Removing original input image: [{input_tag_details.full_tag}]")
+            self.docker_client.images.remove(input_tag_details.full_tag)
             self.docker_client.images.remove(input_tag_details.full_stub_tag)
 
     @backoff.on_exception(backoff.expo, BaseException, max_time=3000) # 50 mins