Fixed error when custom schedule rescaled and how_many_batches > 1

[enzymezoo-code]

Errored during batch runs because the shape variable became a list of floats after the custom schedule resized.

[Dango233]

Thank you for the PR! Will merge this in next release