The definition of ReloadableTransformer was duplicated in reservoir.model to conform with some mypy type constraints. This PR uses cast on the loaded fv3fit.load call because the only saved model for the autoencoder argument must be of that type by construction. This removed the need for that small duplication.
The definition of
ReloadableTransformerwas duplicated inreservoir.modelto conform with some mypy type constraints. This PR usescaston the loadedfv3fit.loadcall because the only saved model for theautoencoderargument must be of that type by construction. This removed the need for that small duplication.