From 02a66202d579a018e33e3eb98b214a69aaa9b3ec Mon Sep 17 00:00:00 2001
From: madhur-ob <155637867+madhur-ob@users.noreply.github.com>
Date: Mon, 7 Oct 2024 23:40:25 +0530
Subject: [PATCH] solve again even if one mismatch (#2083)

---
 metaflow/plugins/pypi/pip.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/metaflow/plugins/pypi/pip.py b/metaflow/plugins/pypi/pip.py
index 5e97f0edd19..13750ea742c 100644
--- a/metaflow/plugins/pypi/pip.py
+++ b/metaflow/plugins/pypi/pip.py
@@ -140,7 +140,10 @@ def download(self, id_, packages, python, platform):
         metadata_file = METADATA_FILE.format(prefix=prefix)
         # download packages only if they haven't ever been downloaded before
         if os.path.isfile(metadata_file):
-            return
+            with open(metadata_file, "r") as file:
+                metadata = json.load(file)
+                if all(package["url"] in metadata for package in packages):
+                    return
 
         metadata = {}
         custom_index_url, extra_index_urls = self.indices(prefix)