model/src/models/visual_communication.py

@@ -1,8 +1,23 @@
 """Definition of VisualCommunicationModel class."""
+
 from __future__ import annotations
 
 from torch import nn
 
+from shared.dto import (
+    AngleData,
+    ContactData,
+    DistanceData,
+    FramingData,
+    InformationValueData,
+    ModalityColorData,
+    ModalityDepthData,
+    ModalityLightingData,
+    PointOfViewData,
+    SalienceData,
+    VisualSyntaxData,
+)
+
 from .angle import AngleTail
 from .contact import ContactTail
 from .distance import DistanceTail
@@ -15,17 +30,6 @@ from .point_of_view import PointOfViewTail
 from .resnet18_head import ResNet18Head
 from .salience import SalienceTail
 from .visual_syntax import VisualSyntaxTail
-from shared.dto import AngleData
-from shared.dto import ContactData
-from shared.dto import DistanceData
-from shared.dto import FramingData
-from shared.dto import InformationValueData
-from shared.dto import ModalityColorData
-from shared.dto import ModalityDepthData
-from shared.dto import ModalityLightingData
-from shared.dto import PointOfViewData
-from shared.dto import SalienceData
-from shared.dto import VisualSyntaxData
 
 
 class VisualCommunicationModel(nn.Module):