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

1""" 

2This module exists only for backward compatibility with object that were pickled by earlier versions of sensAI. 

3""" 

4 

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 

10 

11mark_used(MCDropoutCapableNNModule, MultiLayerPerceptron, LSTNetwork, ResidualFeedForwardNetwork)