from event_extractor.models.heads.head import Head
from event_extractor.models.heads.linear_layer_head import LinearLayerHead
from event_extractor.models.heads.prototypical_head import PrototypicalHead

from event_extractor.models.heads.head import Head
from event_extractor.models.heads.linear_layer_head import LinearLayerHead
from event_extractor.models.heads.prototypical_head import PrototypicalHead
