diff --git a/packages/models-library/src/models_library/basic_types.py b/packages/models-library/src/models_library/basic_types.py index a2b104b049f..3e20878570d 100644 --- a/packages/models-library/src/models_library/basic_types.py +++ b/packages/models-library/src/models_library/basic_types.py @@ -1,5 +1,5 @@ import re -from enum import Enum +from enum import StrEnum from typing import TypeAlias from pydantic import ( @@ -116,14 +116,14 @@ class HttpUrlWithCustomMinLength(HttpUrl): min_length = 0 -class LogLevel(str, Enum): +class LogLevel(StrEnum): DEBUG = "DEBUG" INFO = "INFO" WARNING = "WARNING" ERROR = "ERROR" -class BootModeEnum(str, Enum): +class BootModeEnum(StrEnum): """ Values taken by SC_BOOT_MODE environment variable set in Dockerfile and used during docker/boot.sh @@ -140,7 +140,7 @@ def is_devel_mode(self) -> bool: return self in (self.DEBUG, self.DEVELOPMENT, self.LOCAL) -class BuildTargetEnum(str, Enum): +class BuildTargetEnum(StrEnum): """ Values taken by SC_BUILD_TARGET environment variable set in Dockerfile that defines the stage targeted in the diff --git a/packages/settings-library/src/settings_library/r_clone.py b/packages/settings-library/src/settings_library/r_clone.py index 4bab5f4a327..ff04d509bef 100644 --- a/packages/settings-library/src/settings_library/r_clone.py +++ b/packages/settings-library/src/settings_library/r_clone.py @@ -1,4 +1,4 @@ -from enum import Enum +from enum import StrEnum from pydantic import Field, NonNegativeInt @@ -6,7 +6,7 @@ from .s3 import S3Settings -class S3Provider(str, Enum): +class S3Provider(StrEnum): AWS = "AWS" CEPH = "CEPH" MINIO = "MINIO" diff --git a/services/director-v2/tests/unit/test_core_settings.py b/services/director-v2/tests/unit/test_core_settings.py index 29d3113d155..84d99057f3a 100644 --- a/services/director-v2/tests/unit/test_core_settings.py +++ b/services/director-v2/tests/unit/test_core_settings.py @@ -21,7 +21,7 @@ def _get_backend_type_options() -> set[str]: - return {x.value for x in S3Provider if not x.startswith("_")} + return {x for x in S3Provider if not x.startswith("_")} def test_supported_backends_did_not_change() -> None: