浏览代码

Reintrodute TF for some testing

/develop/rm-rf-new-models
vincentpierre 4 年前
当前提交
89d00673
共有 1 个文件被更改,包括 3 次插入5 次删除
  1. 8
      ml-agents/tests/yamato/yamato_utils.py

8
ml-agents/tests/yamato/yamato_utils.py


pip_commands += [
f"mlagents=={mlagents_python_version}",
f"gym-unity=={mlagents_python_version}",
# TODO build these and publish to internal pypi
"~/tensorflow_pkg/tensorflow-2.0.0-cp37-cp37m-macosx_10_14_x86_64.whl",
"tf2onnx==1.6.1",
if mlagents_python_version == "0.16.0":
# version 0.16.0 of mlagents uses tensorflow only
pip_commands += [
"tensorflow==1.14.0",
]
else:
# Local install
pip_commands += ["-e ./ml-agents-envs", "-e ./ml-agents", "-e ./gym-unity"]

正在加载...
取消
保存