diff --git a/python/fluxqueue/context.py b/python/fluxqueue/context.py index 85b867a..d8686e6 100644 --- a/python/fluxqueue/context.py +++ b/python/fluxqueue/context.py @@ -79,12 +79,7 @@ def __init_subclass__(cls) -> None: if cls.__name__ == "_Context": raise ValueError("Context subclass cannot be named '_Context'") - if cls.__fluxqueue_context__ == "_Context": - raise ValueError( - f"'__fluxqueue_context__' on {cls.__name__} cannot be named '_Context'" - ) - - if not cls.__fluxqueue_context__: + if not cls.__fluxqueue_context__ or cls.__fluxqueue_context__ == "_Context": cls.__fluxqueue_context__ = cls.__name__