Change mixin of str and Enum to StrEnum in NaturalGasCompany.

codeforboston · May 15, 2024 · 9cbff0e · 9cbff0e
1 parent fbf03a0
commit 9cbff0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rules-engine/src/rules_engine/parser.py b/rules-engine/src/rules_engine/parser.py
@@ -5,12 +5,12 @@
 import csv
 import io
 from datetime import datetime, timedelta
-from enum import Enum
+from enum import StrEnum
 
 from .pydantic_models import NaturalGasBillingInput, NaturalGasBillingRecordInput
 
 
-class NaturalGasCompany(str, Enum):
+class NaturalGasCompany(StrEnum):
     EVERSOURCE = "eversource"
     NATIONAL_GRID = "national_grid"