Could not load model due to Error initializing graph component for node

I have updated an existing Rasa 2.0 project to Rasa 3.0 following the migration guide and the custom graph components guide.

My system set up is as follows:

Rasa Version      :         3.0.8
Minimum Compatible Version: 3.0.0
Rasa SDK Version  :         3.0.4
Rasa X Version    :         None
Python Version    :         3.8.3
Operating System  :         Linux-5.4.0-1066-aws-x86_64-with-glibc2.10
Python Path       :         /home/ubuntu/anaconda3/envs/vail-rasa-venv/bin/python

I was able to successfully train a new model. However, when I attempt to run the rasa server with rasa run --enable-api I see the following error in console output (class and module names have been renamed):

Could not load model due to Error initializing graph component for node 'run_nlu.nlu_component.NluComponentClassName2'..

What is confusing to me is that there is no NluComponentClassNamer2 class in the component, but rather just NluComponentClassName. Here are a few snippets of the code in that module. Kindly let me know if more code would be helpful in diagnosing the issue.


    def __init__(
        self,
        component_config: Dict[Text, Any],
        language: Optional[Text] = None,
    ) -> None:
        super().__init__(component_config)
        logger.debug(component_config)
        self.language = language

    @classmethod
    def create(
        cls, 
        config: Dict[Text, Any], 
    ) -> "NluComponentClassName":
        return cls(config)

    @classmethod
    def load(
        cls,
        config: Dict[Text, Any],
        **kwargs: Any,
    ) -> "NluComponentClassName":
        return cls.create(config)

I was able to resolve this issue by removing load function and modeling the init and create functions after the dateparser_extractor example here.