code: def get_task_templates():
      task_templates = {}
      task_templates["task name A"] = {
          "prompt_template": "<|VIRTUAL_PROMPT_0|>{text}{answer}",
          "prompt_template_fields": ["text", "answer"],
          "total_virtual_tokens": 5,
          "virtual_token_splits": [5],
          "truncate_field": None,
          "answer_only_loss": True,
          "answer_field": "answer",
          "task_id_num": 0,
      }
      task_templates["task name B"] = {
          "prompt_template": "<|VIRTUAL_PROMPT_0|>{question}<|VIRTUAL_PROMPT_1|>{answer}{extra}",
          "prompt_template_fields": ["question", "answer", "extra"],
          "total_virtual_tokens": 10,
          "virtual_token_splits": [7, 3],
          "truncate_field": None,
          "answer_only_loss": False,
          "answer_field": None,
          "task_id_num": 1,
      }
      return task_templates
  
file_path: NeMo/tests/collections/nlp/test_prompt_learning.py
imports: import json
  import os
  import pytest
  import torch
  from nemo.collections.nlp.data.language_modeling.megatron.gpt_prompt_learning_dataset import GPTPromptLearningDataset
  from nemo.collections.nlp.models.language_modeling.megatron_gpt_prompt_learning_model import get_pseudo_tokens
  from nemo.collections.nlp.modules.common import VirtualPromptSource
  from nemo.collections.nlp.modules.common.tokenizer_utils import get_nmt_tokenizer
  from nemo.core import Dataset
label_desc: Use image node_id 3 for calling a global function with example usage: get_task_templates() and returns: task_templates