I'm registering a task and it is getting registered at some other place. module_name<http://wf_v1.wf|.<>workflow_name>.flytekitplugins.kfpytorch.task.module_name.task_name. Can someone help me in registering it as module_name.task_name? flytekitplugins.kfpytorch.task this term is coming due to the use of torch elastic in the task invocation. But according to my knowledge it should be implicit. Due to the same I'm getting mapping out of range error telling task is not registered at the correct location, while load task is called.

would you mind sharing the directory structure and code if possible?