| """Compatibility exports for the official ClimODE checkpoint pickle.""" | |
| from model.climode import ( | |
| ClimateEncoderFreeUncertain, | |
| ClimateResNet2D, | |
| Climate_encoder_free_uncertain, | |
| Climate_ResNet_2D, | |
| OptimVelocity, | |
| SelfAttnConv, | |
| Self_attn_conv, | |
| ) | |
| __all__ = [ | |
| "Climate_encoder_free_uncertain", | |
| "Climate_ResNet_2D", | |
| "OptimVelocity", | |
| "Self_attn_conv", | |
| "ClimateEncoderFreeUncertain", | |
| "ClimateResNet2D", | |
| "SelfAttnConv", | |
| ] | |