Add missing import.

kpfleming · kpfleming · commit ca82f47df049 · 2024-07-04T09:34:23.000-04:00
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -125,6 +125,7 @@
     ValveState,
     VoiceAssistantFeature,
     WaterHeaterInfo,
+    WaterHeaterState,
     build_unique_id,
     converter_field,
 )
