Unable to generate the weather using generate.py with time Series training checkpoint

[RUOWANJI]

The repeat function in generate.py can only take the dataset with two dimensions, but the weather-dataset has three dimensions, I've tried change the size of repeating but it still doesn't work.