[Cherry Pick] [Fix] Add more verbosewandb import error (#1375)

* clean cherry pick

* new solution

* final fix

src/sparseml/pytorch/utils/logger.py

@@ -534,7 +534,14 @@ def __init__(
         init_kwargs: Optional[Dict] = None,
         name: str = "wandb",
         enabled: bool = True,
+        wandb_err: Optional[Exception] = wandb_err,
     ):
+        if wandb_err:
+            raise ModuleNotFoundError(
+                "Error: Failed to import wandb. "
+                "Please install the wandb library in order to use it."
+            ) from wandb_err
+
         super().__init__(
             lambda_func=self._log_lambda,
             name=name,