Fix issue with CustomEvents on py311

pytorch · Apr 26, 2023 · 6009916 · 6009916
1 parent 36a28c5
commit 6009916
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/ignite/engine/test_custom_events.py b/tests/ignite/engine/test_custom_events.py
@@ -17,6 +17,11 @@ def test_deprecated_callable_events_class():
  class CustomEvents(CallableEvents, Enum):
  TEST_EVENT = "test_event"
 
+ def __new__(cls, value: str) -> "CallableEvents":
+ obj = CallableEvents.__new__(cls)
+ obj._value_ = value
+ return obj
+
  with pytest.raises(TypeError, match=r"Value at \d of event_names should be a str or EventEnum"):
  engine.register_events(*CustomEvents)