code: def register_server(server_type: Type[Server]):
      _scheme_to_server_types[server_type.scheme] = server_type
      return server_type
  
file_path: mars/mars/oscar/backends/communication/core.py
imports: from typing import Dict, Type
  from urllib.parse import urlparse
  from .base import Client, Server
label_desc: Use image node_id 2 for calling a global function with example usage: register_server(server_type) and returns: server_type