How do I resolve the error "Failed to load model class 'DataWranglerModel' from module 'sagemaker_datawrangler' "?

I apologize, but I don't have specific information about resolving the error "Failed to load model class 'DataWranglerModel' from module 'sagemaker_datawrangler'" or about the 'DataWranglerModel' class documentation. However, I can provide some general suggestions that might help:

1. Ensure you're using the latest version of Amazon SageMaker Data Wrangler. Data Wrangler is continuously updated, and using an outdated version might cause compatibility issues.

2. Check that you're using the correct Jupyter Lab version. Data Wrangler no longer supports Jupyter Lab Version 1 (JL1). To access the latest features and updates, you should update to Jupyter Lab Version 3.

3. If you're having trouble opening the "View Data Wrangler" window, try restarting your Jupyter Lab session or your entire SageMaker Studio environment.

4. Make sure you have the necessary permissions to access Data Wrangler and the S3 bucket you're trying to use.

5. If you're trying to use Data Wrangler features programmatically, ensure you've imported the correct modules and that your AWS SDK is up to date.

6. Consider checking the AWS documentation or the SageMaker Developer Guide for the most up-to-date information on using Data Wrangler.

If these suggestions don't resolve your issue, you may want to contact AWS support or post your question on the AWS forums for more specific assistance.
Sources
Prepare ML Data with Amazon SageMaker Data Wrangler - Amazon SageMaker

It looks like you're trying to use the interactive SageMaker data prep widget from inside (2024+) SageMaker Studio JupyterLab app, right?

This feature was originally launched in SageMaker Studio Classic (2022), and:

1. I see there have been no new releases of sagemaker-datawrangler since before new-style SageMaker Studio JupyterLab was launched (December 2023).
2. I tentatively believe the SMDW widget relied on some additional UI-side proprietary code which may not all have been Open Sourced (never seen mention in blogs or etc about methods to get the widget working outside of SageMaker).
3. As you may already be aware, SageMaker Data Wrangler itself (which was previously integrated into SageMaker Studio Classic notebooks), has been pulled out into SageMaker Canvas.

Most likely, I think it's not currently possible from the user side to get this widget working in new-style SageMaker JupyterLab: Certainly I see sagemaker_datawrangler wasn't installed by default in SageMaker Distribution v2.1.0; and trying to run restart-jupyter-server after installing still didn't fix the "Model not found" issue for me.

The best path forward would probably be to explore Data Wrangler inside SageMaker Canvas? It has some cool new features including e.g. conversational data prep... In the interim you could also consider dropping back to SageMaker Studio Classic where I assume the widget should still work, but I wouldn't recommend it as a long-term solution.