text: #!/usr/bin/env python3
  
  # Copyright (c) Facebook, Inc. and its affiliates.
  #
  # This source code is licensed under the MIT license found in the
  # LICENSE file in the root directory of this source tree.
  
  """Dataset paths."""
  
  import os
  
  # Default data directory (/path/pycls/pycls/datasets/data)
  _DEF_DATA_DIR = os.path.join(os.path.dirname(__file__), 'data')
  
  # Data paths
  _paths = {
      'cifar10': _DEF_DATA_DIR + '/cifar10',
      'imagenet': _DEF_DATA_DIR + '/imagenet'
  }
  
  
  def has_data_path(dataset_name):
      """Determines if the dataset has a data path."""
      return dataset_name in _paths.keys()
  
  
  def get_data_path(dataset_name):
      """Retrieves data path for the dataset."""
      return _paths[dataset_name]
  
  
  def set_data_path(dataset_name, data_path):
      """Sets data path for the dataset."""
      _paths[dataset_name] = data_path
  