benchmarking/repos/repos.py (11 lines of code) (raw):
#!/usr/bin/env python
##############################################################################
# Copyright 2017-present, Facebook, Inc.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
##############################################################################
from __future__ import absolute_import, division, print_function, unicode_literals
from .git import GitRepo
from .hg import HGRepo
def getRepo(repo, repo_dir):
if repo == "git":
return GitRepo(repo_dir)
elif repo == "hg":
return HGRepo(repo_dir)
else:
raise AssertionError("Repo not recognized")
return None