class_docstr: Implementation class of an MII inference server
class_docstr_tok: ["Implementation","class","of","an","MII","inference","server"]
code: def _get_task_methods(
      self, method_name: str
  ) -> Dict[str, TaskMethods]:
      if method_name not in self.method_name_to_task:
          raise ValueError(f"unknown method: {method_name}")
  
      task = self.method_name_to_task[method_name]
      if task not in TASK_METHODS_DICT:
          raise ValueError(f"unknown task: {task}")
  
      task_methods = TASK_METHODS_DICT[task]
      return task_methods
  
imports: import asyncio
  import queue
  import sys
  import threading
  from concurrent import futures
  from typing import Dict
  import grpc
  from google.protobuf import empty_pb2
  from mii.backend.client import create_channel
  from mii.constants import GenerationFinishReason, GRPC_MAX_MSG_SIZE, TERMINATE_METHOD, LB_MAX_WORKER_THREADS, SERVER_SHUTDOWN_TIMEOUT, STREAM_RESPONSE_QUEUE_TIMEOUT
  from mii.grpc_related.proto import modelresponse_pb2_grpc
  from mii.grpc_related.task_methods import TASK_METHODS_DICT, TaskMethods
label_desc: Use image node_id 2 for calling the ModelResponse obj's underlying member method code with example usage: obj._get_task_methods(method_name) and returns: task_methods