Merge pull request #4 from neulab/main

Merge from main.
Anindyadeep · Jan 3, 2024 · 2a87534 · 2a87534
2 parents 745ee05 + abcb7dc
commit 2a87534
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/prompt2model/model_retriever/description_based_retriever.py b/prompt2model/model_retriever/description_based_retriever.py
@@ -131,11 +131,10 @@ def load_model_info(self):
         """
         if not os.path.isdir(self.model_descriptions_index_path):
             # If the model descriptions directory is not populated, then populate it.
-            urllib.request.urlretrieve(
-                "http://phontron.com/data/prompt2model/model_info.tgz",
-                "/tmp/model_info.tgz",
+            temporary_file, _ = urllib.request.urlretrieve(
+                "http://phontron.com/data/prompt2model/model_info.tgz"
             )
-            tar = tarfile.open("/tmp/model_info.tgz")
+            tar = tarfile.open(temporary_file)
             os.makedirs(self.model_descriptions_index_path)
             tar.extractall(path=self.model_descriptions_index_path)