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()