code: def get_slurm_partition(
      cluster_type: Optional[ClusterType] = None,
  ) -> Optional[str]:
      cluster_type = get_cluster_type(cluster_type)
      if cluster_type is None:
          return None
  
      SLURM_PARTITIONS = {
          ClusterType.AWS: "learnlab",
          ClusterType.FAIR: "learnlab",
          ClusterType.RSC: "learn",
      }
      return SLURM_PARTITIONS[cluster_type]
  
file_path: dinov2/dinov2/utils/cluster.py
imports: from enum import Enum
  import os
  from pathlib import Path
  from typing import Any, Dict, Optional
label_desc: Use image node_id 5 for calling a global function with example usage: get_slurm_partition(cluster_type) and returns: SLURM_PARTITIONS, None