fix error when config class is none

lenskit · Jan 14, 2025 · dec445c · dec445c
1 parent 38f6aaa
commit dec445c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lenskit/lenskit/pipeline/components.py b/lenskit/lenskit/pipeline/components.py
@@ -14,7 +14,7 @@
 from abc import ABC, abstractmethod
 from importlib import import_module
 from inspect import isabstract
-from types import FunctionType
+from types import FunctionType, NoneType
 
 from pydantic import JsonValue, TypeAdapter
 from typing_extensions import (
@@ -169,7 +169,7 @@ def __init__(self, config: object | None = None, **kwargs: Any):
     def _config_class(cls, return_any: bool = False) -> type | None:
         hints = get_type_hints(cls)
         ct = hints.get("config", None)
-        if ct is None:
+        if ct is None or ct == NoneType:
             return None
         elif ct == Any:
             if return_any: