fix: <3.12 python support enum in

Author: tazlin

horde_sdk/ai_horde_api/apimodels/generate/pop.py

@@ -381,7 +381,7 @@ def has_upscaler(self) -> bool:
             return False
 
         return any(
-            post_processing in KNOWN_UPSCALERS.__members__ or post_processing in KNOWN_UPSCALERS._value2member_map_
+            post_processing in KNOWN_UPSCALERS.__members__ or post_processing in KNOWN_UPSCALERS.__members__.values()
             for post_processing in self.payload.post_processing
         )
 

horde_sdk/generation_parameters/alchemy/consts.py

@@ -144,7 +144,7 @@ def is_upscaler_form(form: KNOWN_ALCHEMY_TYPES | str) -> bool:
     if isinstance(form, KNOWN_ALCHEMY_TYPES):
         value = form.value
 
-    return value in KNOWN_UPSCALERS or value in KNOWN_UPSCALERS.__members__.values()
+    return value in KNOWN_UPSCALERS.__members__ or value in KNOWN_UPSCALERS.__members__.values()
 
 
 def is_facefixer_form(form: KNOWN_ALCHEMY_TYPES | str) -> bool:
@@ -153,7 +153,7 @@ def is_facefixer_form(form: KNOWN_ALCHEMY_TYPES | str) -> bool:
     if isinstance(form, KNOWN_ALCHEMY_TYPES):
         value = form.value
 
-    return value in KNOWN_FACEFIXERS or value in KNOWN_FACEFIXERS.__members__.values()
+    return value in KNOWN_FACEFIXERS.__members__ or value in KNOWN_FACEFIXERS.__members__.values()
 
 
 def is_interrogator_form(form: KNOWN_ALCHEMY_TYPES | str) -> bool:

tests/ai_horde_api/test_ai_horde_api_models.py

@@ -433,7 +433,7 @@ def test_ImageGenerateJobPopResponse() -> None:
     )
 
     assert all(
-        post_processor in KNOWN_UPSCALERS._value2member_map_
+        post_processor in KNOWN_UPSCALERS.__members__.keys() or post_processor in KNOWN_UPSCALERS.__members__.values()
         for post_processor in test_response.payload.post_processing
     )
 
@@ -448,7 +448,10 @@ def test_ImageGenerateJobPopResponse() -> None:
         skipped=ImageGenerateJobPopSkippedStatus(),
     )
 
-    assert all(post_processor in KNOWN_UPSCALERS for post_processor in test_response.payload.post_processing)
+    assert all(
+        post_processor in KNOWN_UPSCALERS.__members__ or post_processor in KNOWN_UPSCALERS.__members__.values()
+        for post_processor in test_response.payload.post_processing
+    )
 
     test_response = ImageGenerateJobPopResponse(
         ids=[

tests/worker/dispatch/ai_horde/test_alchemy_parameters.py

@@ -52,19 +52,34 @@ def assert_common_parameters(
 
         if isinstance(single_generation, UpscaleAlchemyParameters):
             assert single_generation.form == KNOWN_ALCHEMY_FORMS.post_process
-            assert single_generation.upscaler in KNOWN_UPSCALERS
+            assert (
+                single_generation.upscaler in KNOWN_UPSCALERS.__members__
+                or single_generation.upscaler in KNOWN_UPSCALERS.__members__.values()
+            )
         elif isinstance(single_generation, FacefixAlchemyParameters):
             assert single_generation.form == KNOWN_ALCHEMY_FORMS.post_process
-            assert single_generation.facefixer in KNOWN_FACEFIXERS
+            assert (
+                single_generation.facefixer in KNOWN_FACEFIXERS.__members__
+                or single_generation.facefixer in KNOWN_FACEFIXERS.__members__.values()
+            )
         elif isinstance(single_generation, InterrogateAlchemyParameters):
             assert single_generation.form == KNOWN_ALCHEMY_FORMS.interrogation
-            assert single_generation.interrogator in KNOWN_INTERROGATORS
+            assert (
+                single_generation.interrogator in KNOWN_INTERROGATORS.__members__
+                or single_generation.interrogator in KNOWN_INTERROGATORS.__members__.values()
+            )
         elif isinstance(single_generation, CaptionAlchemyParameters):
             assert single_generation.form == KNOWN_ALCHEMY_FORMS.caption
-            assert single_generation.caption_model in KNOWN_CAPTION_MODELS
+            assert (
+                single_generation.caption_model in KNOWN_CAPTION_MODELS.__members__
+                or single_generation.caption_model in KNOWN_CAPTION_MODELS.__members__.values()
+            )
         elif isinstance(single_generation, NSFWAlchemyParameters):
             assert single_generation.form == KNOWN_ALCHEMY_FORMS.nsfw
-            assert single_generation.nsfw_detector in KNOWN_NSFW_DETECTOR
+            assert (
+                single_generation.nsfw_detector in KNOWN_NSFW_DETECTOR.__members__
+                or single_generation.nsfw_detector in KNOWN_NSFW_DETECTOR.__members__.values()
+            )
 
 
 def test_convert_alchemy_job_pop_response_to_parameters_interrogate(