File size: 522 Bytes
05c9ac2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import argparse
from .yamato_utils import init_venv
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--mlagents-version", default=None)
parser.add_argument("--extra-packages", default=None)
args = parser.parse_args()
extra_packages = []
if args.extra_packages is not None:
extra_packages = args.extra_packages.split(",")
init_venv(
mlagents_python_version=args.mlagents_version, extra_packages=extra_packages
)
if __name__ == "__main__":
main()
|