Add all models to init all

Author: mdegat01

aiohasupervisor/models/__init__.py

@@ -23,6 +23,21 @@
     StoreInfo,
     SupervisorRole,
 )
+from aiohasupervisor.models.os import (
+    BootSlot,
+    BootSlotName,
+    DataDisk,
+    DataDiskList,
+    GreenInfo,
+    GreenOptions,
+    MigrateDataOptions,
+    OSInfo,
+    OSUpdate,
+    RaucState,
+    SetBootSlotOptions,
+    YellowInfo,
+    YellowOptions,
+)
 from aiohasupervisor.models.resolution import (
     Check,
     CheckOptions,
@@ -86,4 +101,17 @@
     "SuggestionType",
     "UnhealthyReason",
     "UnsupportedReason",
+    "RaucState",
+    "BootSlotName",
+    "BootSlot",
+    "OSInfo",
+    "OSUpdate",
+    "MigrateDataOptions",
+    "DataDisk",
+    "DataDiskList",
+    "SetBootSlotOptions",
+    "GreenInfo",
+    "GreenOptions",
+    "YellowInfo",
+    "YellowOptions",
 ]

tests/test_os.py

@@ -4,7 +4,7 @@
 from yarl import URL
 
 from aiohasupervisor import SupervisorClient
-from aiohasupervisor.models.os import (
+from aiohasupervisor.models import (
     BootSlotName,
     GreenOptions,
     MigrateDataOptions,