Coverage for src/sensai/torch/torch_modules.py: 100%
6 statements
« prev ^ index » next coverage.py v7.6.1, created at 2024-08-13 22:17 +0000
« prev ^ index » next coverage.py v7.6.1, created at 2024-08-13 22:17 +0000
1"""
2This module exists only for backward compatibility with object that were pickled by earlier versions of sensAI.
3"""
5from .torch_base import MCDropoutCapableNNModule
6from .torch_models.mlp.mlp_modules import MultiLayerPerceptron
7from .torch_models.lstnet.lstnet_modules import LSTNetwork
8from .torch_models.residualffn.residualffn_modules import ResidualFeedForwardNetwork
9from ..util import mark_used
11mark_used(MCDropoutCapableNNModule, MultiLayerPerceptron, LSTNetwork, ResidualFeedForwardNetwork)