Fix TrainStopStatus

Author: gjohansson-ST

pytrafikverket/trafikverket_train.py

@@ -99,20 +99,20 @@ def __init__(
     def get_state(self) -> TrainStopStatus:
         """Retrieve the state of the departure."""
         if self.canceled:
-            return TrainStopStatus.CANCELED.value
+            return TrainStopStatus.CANCELED
         if (
             self.advertised_time_at_location is not None
             and self.time_at_location is not None
             and self.advertised_time_at_location != self.time_at_location
         ):
-            return TrainStopStatus.DELAYED.value
+            return TrainStopStatus.DELAYED
         if (
             self.advertised_time_at_location is not None
             and self.estimated_time_at_location is not None
             and self.advertised_time_at_location != self.estimated_time_at_location
         ):
-            return TrainStopStatus.DELAYED.value
-        return TrainStopStatus.ON_TIME.value
+            return TrainStopStatus.DELAYED
+        return TrainStopStatus.ON_TIME
 
     def get_delay_time(self) -> timedelta | None:
         """Calculate the delay of a departure."""

setup.py

@@ -7,7 +7,7 @@
 
 setup(
     name="pytrafikverket",
-    version="0.3.2",
+    version="0.3.3",
     description="Retreive values from public API at the Swedish Transport Administration (Trafikverket).",
     long_description=long_description,
     long_description_content_type="text/markdown",