Using interleave_datasets with probabilities

Hi,

I’m trying to use the interleave_datasets function with probabilities. Now, my use case some probabilities might be 0. For example, probabilities=[0.5, 0.5, 0] where I would like to sample from the first and second dataset at 50% likelihood, and never from the third.

Reading the documentation I figured this would be supported. But when trying it, the code seems to loop forever. Does anyone have any experience with this, or some more knowledge of the inner workings that could clear this up?

Thanks in advance!