1234567891011121314151617181920212223242526272829303132333435 |
- import os
- import ray
- import sys
- RAY_VERSION = "RAY_VERSION"
- RAY_COMMIT = "RAY_HASH"
- ray_version = os.getenv(RAY_VERSION)
- ray_commit = os.getenv(RAY_COMMIT)
- if __name__ == "__main__":
- print("Sanity check python version: {}".format(sys.version))
- assert (
- ray_version == ray.__version__
- ), "Given Ray version {} is not matching with downloaded " "version {}".format(
- ray_version, ray.__version__
- )
- assert (
- ray_commit == ray.__commit__
- ), "Given Ray commit {} is not matching with downloaded " "version {}".format(
- ray_commit, ray.__commit__
- )
- assert ray.__file__ is not None
- ray.init()
- assert ray.is_initialized()
- @ray.remote
- def return_arg(arg):
- return arg
- val = 3
- print("Running basic sanity check.")
- assert ray.get(return_arg.remote(val)) == val
- ray.shutdown()
|