Skip to content

Commit

Permalink
Updated training init tests to remove inference test temporarily. (#5984
Browse files Browse the repository at this point in the history
)
  • Loading branch information
miguelalonsojr authored Oct 5, 2023
1 parent 4e39553 commit ddcc42a
Showing 1 changed file with 8 additions and 6 deletions.
14 changes: 8 additions & 6 deletions ml-agents/tests/yamato/training_int_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,12 +105,14 @@ def run_training(python_version: str, csharp_version: str) -> bool:
subprocess.run(["cat", log_output_path])
return False

if csharp_version is None and python_version is None:
# Use abs path so that loading doesn't get confused
model_path = os.path.abspath(os.path.dirname(onnx_file_expected))
inference_ok = run_inference(env_path, model_path, "onnx")
if not inference_ok:
return False
# TODO re-enable when Sentis model byte loading has been added
# disabling inference tests for now due to Sentis deprecation of loading onnx from byte array.
# if csharp_version is None and python_version is None:
# # Use abs path so that loading doesn't get confused
# model_path = os.path.abspath(os.path.dirname(onnx_file_expected))
# inference_ok = run_inference(env_path, model_path, "onnx")
# if not inference_ok:
# return False

print("mlagents-learn run SUCCEEDED!")
return True
Expand Down

0 comments on commit ddcc42a

Please sign in to comment.