sigs.k8s.io/kueue@v0.6.2/site/static/examples/jobs/ray-job-code-sample.yaml (about)

     1  apiVersion: v1
     2  kind: ConfigMap
     3  metadata:
     4    name: ray-job-code-sample
     5  data:
     6    sample_code.py: |
     7      import ray
     8      import os
     9      import requests
    10  
    11      ray.init()
    12  
    13      @ray.remote
    14      class Counter:
    15          def __init__(self):
    16              # Used to verify runtimeEnv
    17              self.name = os.getenv("counter_name")
    18              self.counter = 0
    19  
    20          def inc(self):
    21              self.counter += 1
    22  
    23          def get_counter(self):
    24              return "{} got {}".format(self.name, self.counter)
    25  
    26      counter = Counter.remote()
    27  
    28      for _ in range(5):
    29          ray.get(counter.inc.remote())
    30          print(ray.get(counter.get_counter.remote()))
    31  
    32      print(requests.__version__)