from __future__ import annotations from .fully_connected import FullyConnectedModel class InformationValueTail(FullyConnectedModel): def __init__(self): super().__init__(num_out_features=3)