New tools for Nonprofits: Community Training and Likelihood to Donate

I'm curious if anyone else has seen the flow failure error we're experiencing with the Likelihood to Donate functionality?

We've followed the instructions here (https://learn.microsoft.com/en-us/dynamics365/industry/nonprofit/fundraising-engagement-likelihood-to-donate) but when we get to the "Train the model and generate categories" section and try to do steps 4-6, the flow fails every time.

The failure is on the last step which is called "Checking hydration status" -- see screenshot below. We'd appreciate any assistance troubleshooting this - thanks!