def _entry_script_path()

in launcher/nemo/recipe_stages.py [0:0]

• 6 lines of code
• 2 McCabe index (conditional complexity)

    def _entry_script_path(self) -> Path:
        # [TODO] Handle generate the script path from github
        choice_model_type, _ = self.get_stage_config_choice()
        choice_model_type = choice_model_type.split("/")[1]
        # predefined model
        if choice_model_type in SM_ADAPTER_MODEL_TYPE_TO_CODE_PATH:
            return Path(SM_ADAPTER_MODEL_TYPE_TO_CODE_PATH[choice_model_type])
        # custom model
        return Path("examples/custom_model/custom_pretrain.py")