info traces

ServiceNow · Oct 30, 2024 · 00e0c6b · 00e0c6b
1 parent 508deee
commit 00e0c6b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/browsergym/experiments/src/browsergym/experiments/benchmark/base.py b/browsergym/experiments/src/browsergym/experiments/benchmark/base.py
@@ -163,6 +163,7 @@ def prepare_backends(self):
                     metadata_path = cache_dir / "metadata.json"
 
                     if not metadata_path.exists():
+                        logger.info(f"Downloading metadata.json")
                         huggingface_hub.snapshot_download(
                             repo_id="McGill-NLP/weblinx-browsergym",
                             repo_type="dataset",
@@ -183,13 +184,13 @@ def prepare_backends(self):
                                 if not base_demo_dir.joinpath(demo_id).exists():
                                     # first, if zip file does not exist, download the zip file
                                     if not base_zip_dir.joinpath(f"{demo_id}.zip").exists():
+                                        logger.info(f"Downloading demonstrations_zip/{demo_id}.zip")
                                         huggingface_hub.snapshot_download(
                                             repo_id="McGill-NLP/weblinx-browsergym",
                                             repo_type="dataset",
                                             local_dir=cache_dir,
                                             allow_patterns=[f"demonstrations_zip/{demo_id}.zip"],
                                         )
-                                        logger.debug(f"Downloaded {demo_id}.zip to {base_zip_dir}")
 
                                     # then, unzip the file
                                     with zipfile.ZipFile(