fix: PDB_MIRROR variable error catching

Nathaniel Corley · Nathaniel Corley · commit 1ba98a9a52a9 · 2025-08-15T18:27:18.000+02:00
diff --git a/src/modelhub/utils/datasets.py b/src/modelhub/utils/datasets.py
@@ -30,9 +30,9 @@
         FallbackDatasetWrapper,
         get_row_and_index_by_example_id,
     )
-except (ImportError, ModuleNotFoundError):
+except Exception as e:
     ranked_logger.warning(
-        "Failed to import atomworks.ml.datasets.datasets. "
+        f"Failed to import atomworks.ml.datasets.datasets: {type(e).__name__}: {e}. "
         "If training networks, the PDB_MIRROR environment variable must be set."
     )
 

Original file line number	Diff line number	Diff line change
`@@ -30,9 +30,9 @@`
`30`	`30`	`FallbackDatasetWrapper,`
`31`	`31`	`get_row_and_index_by_example_id,`
`32`	`32`	`)`
`33`		`-except (ImportError, ModuleNotFoundError):`
	`33`	`+except Exception as e:`
`34`	`34`	`ranked_logger.warning(`
`35`		`- "Failed to import atomworks.ml.datasets.datasets. "`
	`35`	`+ f"Failed to import atomworks.ml.datasets.datasets: {type(e).__name__}: {e}. "`
`36`	`36`	`"If training networks, the PDB_MIRROR environment variable must be set."`
`37`	`37`	`)`
`38`	`38`