TypeError: Subscripted generics cannot be used with class and instance checks

[kbaltakys]

trafficstars

isinstance(device_train_microbatch_size, Union[int, float]) should be isinstance(device_train_microbatch_size, (int, float)) on line https://github.com/mosaicml/composer/blob/3c0a8179d2c3d60786226a73d8f935bc46306a31/composer/trainer/trainer.py#L298