mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
30 lines
907 B
Python
30 lines
907 B
Python
|
"""Multi-GPU learning tests for RLlib (torch and tf).
|
||
|
|
"""
|
||
|
|
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from ray.rllib.utils.test_utils import run_learning_tests_from_yaml
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
# Get path of this very script to look for yaml files.
|
||
|
|
abs_yaml_path = Path(__file__).parent
|
||
|
|
print("abs_yaml_path={}".format(abs_yaml_path))
|
||
|
|
|
||
|
|
# This pattern match is kind of hacky. Avoids cluster.yaml to get sucked
|
||
|
|
# into this.
|
||
|
|
yaml_files = abs_yaml_path.rglob("*test*.yaml")
|
||
|
|
yaml_files = sorted(
|
||
|
|
map(lambda path: str(path.absolute()), yaml_files), reverse=True)
|
||
|
|
|
||
|
|
# Run all tests in the found yaml files.
|
||
|
|
results = run_learning_tests_from_yaml(yaml_files)
|
||
|
|
|
||
|
|
test_output_json = os.environ.get(
|
||
|
|
"TEST_OUTPUT_JSON", "/tmp/rllib_multi_gpu_learning_tests.json")
|
||
|
|
with open(test_output_json, "wt") as f:
|
||
|
|
json.dump(results, f)
|
||
|
|
|
||
|
|
print("Ok.")
|