Manually wrap pip errors to help remind users to install requirements…

….txt
rfbgo · Jun 7, 2024 · d81f65d · d81f65d
1 parent f7d2225
commit d81f65d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/lib/ramble/ramble/graphs.py b/lib/ramble/ramble/graphs.py
@@ -103,7 +103,11 @@ def walk(self):
             node (GraphNode): Each node in the graph
         """
         if not self._prepared:
-            sorter = graphlib.TopologicalSorter(self.adj_list)
+            try:
+                sorter = graphlib.TopologicalSorter(self.adj_list)
+            except AttributeError as e:
+                logger.die("graphlib.TopologicalSorter is not found. Ensure requirements.txt are installed (including backports, where needed).")
+
             try:
                 self._sorted = tuple(sorter.static_order())
             except graphlib.CycleError as e:

diff --git a/lib/ramble/ramble/pipeline.py b/lib/ramble/ramble/pipeline.py
@@ -150,12 +150,15 @@ def _execute(self):
                 or self.suppress_per_experiment_prints
             )
             if not disable_progress:
-                progress = tqdm.tqdm(
-                    total=len(phase_list),
-                    leave=True,
-                    ascii=" >=",
-                    bar_format="{l_bar}{bar}| Elapsed (s): {elapsed_s:.2f}",
-                )
+                try:
+                    progress = tqdm.tqdm(
+                        total=len(phase_list),
+                        leave=True,
+                        ascii=" >=",
+                        bar_format="{l_bar}{bar}| Elapsed (s): {elapsed_s:.2f}",
+                    )
+                except AttributeError:
+                    logger.die("tdqm.tdqm is not found. Ensure requirements.txt are installed.")
             for phase_idx, phase in enumerate(phase_list):
                 if not disable_progress:
                     progress.set_description(