code: def get_taskmodel_kwargs(self, taskmodel_name: str) -> Optional[Dict]:
      assert taskmodel_name in self.task_to_taskmodel_map.values()
      if self.taskmodel_config_map is None:
          return None
      elif taskmodel_name not in self.taskmodel_config_map:
          return None
      else:
          return self.taskmodel_config_map[taskmodel_name]
  
imports: import warnings
  from dataclasses import dataclass
  from typing import Dict, List, Optional
  import jiant.proj.main.components.task_sampler
  import jiant.shared.caching
  from jiant.tasks.core import Task
  from jiant.tasks.retrieval import create_task_from_config_path
  import jiant.utils.python.io
  from jiant.utils.python.datastructures import ExtendedDataClassMixin
label_desc: Use image node_id 1 for calling the TaskmodelsConfig obj's underlying member method code with example usage: obj.get_taskmodel_kwargs(taskmodel_name) and returns: None, None, self