fix: PDB_MIRROR variable error catching

This commit is contained in:
Nathaniel Corley
2025-08-15 18:26:20 +02:00
parent e40c5c2168
commit 1ba98a9a52

View File

@@ -30,9 +30,9 @@ try:
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."
)