print before loading database

snap-stanford · Nov 27, 2023 · 4380ff4 · 4380ff4
1 parent b1746fd
commit 4380ff4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/relbench/data/dataset.py b/relbench/data/dataset.py
@@ -69,7 +69,11 @@ def __init__(self, *, process: bool = False) -> None:
                 processor=unzip_processor,
                 progressbar=True,
             )
+            print(f"loading Database object from {db_path}...")
+            tic = time.time()
             db = Database.load(db_path)
+            toc = time.time()
+            print(f"done in {toc - tic:.2f} seconds.")
 
         super().__init__(
             db, self.val_timestamp, self.test_timestamp, self.task_cls_list