code: def azure_local_ssd_size(worker_type: str, cores: int) -> int:
      return int(
          cores
          * azure_local_ssd_size_per_core_by_worker_type[worker_type]
      )
  
imports: import logging
  import re
  from typing import Dict, Optional, Tuple
  import sortedcontainers
  from ...globals import RESERVED_STORAGE_GB_PER_CORE
label_desc: Use image node_id 6 for calling a global function with example usage: azure_local_ssd_size(worker_type, cores) and returns: int