fixed test

nens · Nov 26, 2024 · dad6b44 · dad6b44
1 parent 31da8d5
commit dad6b44
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/threedigrid/admin/structure_controls/models.py b/threedigrid/admin/structure_controls/models.py
@@ -1,4 +1,4 @@
-from enum import Enum, StrEnum
+from enum import Enum
 from typing import List
 
 
@@ -8,7 +8,7 @@ class StructureControlTypes(Enum):
     timed_control = "timed_control"
 
 
-class StructureControlSourceTypes(StrEnum):
+class StructureControlSourceTypes(str, Enum):
     LINES = "lines"
     NODES = "nodes"
     PUMPS = "pumps"