def get_root_dir(name):
  """Return the corresponding root dir given taks name.

  Args:
    name: task name

  Returns:
    root dir
  """
  root_dir = FLAGS.root_dir
  if name == 'metaqa2':
    root_dir += 'MetaQA/2hop/'
  elif name == 'metaqa3':
    root_dir += 'MetaQA/3hop/'
  elif name == 'webqsp':
    root_dir += 'WebQSP/'
  elif name in ['membership', 'intersection', 'union', 'set_follow', 'mixture']:
    root_dir += '/'
  elif name.startswith('query2box'):
    root_dir += '/'
  else:
    raise ValueError('name not recognized')
  return root_dir
Return the corresponding root dir given taks name.

  Args:
    name: task name

  Returns:
    root dir
  
language/emql/main.py