def _retrieve_class_or_fn(
    name, registered_name, module, obj_type, full_config, custom_objects=None
):
     ## ... truncated for brevity...
        package = module.split(".", maxsplit=1)[0]
        if package in {"keras", "keras_hub", "keras_cv", "keras_nlp"}:
            try:
                mod = importlib.import_module(module)
                obj = vars(mod).get(name, None)
                if obj is not None:
                    return obj
            except ModuleNotFoundError:
                raise TypeError(..)

    raise TypeError(...)