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