diff --git a/callbacks/app.py b/callbacks/app.py index 5fb3eab..1ae4fce 100644 --- a/callbacks/app.py +++ b/callbacks/app.py @@ -1,7 +1,7 @@ from beam import function -@function(callback_url="https://www.beam.cloud/") +@function(app="examples", callback_url="https://www.beam.cloud/") def handler(x): return {"result": x} diff --git a/custom_images/app.py b/custom_images/app.py index 0c3d132..0369892 100644 --- a/custom_images/app.py +++ b/custom_images/app.py @@ -11,7 +11,7 @@ ) -@endpoint(image=image) +@endpoint(app="examples", image=image) def handler(): import torch diff --git a/endpoints/app.py b/endpoints/app.py index a0803d3..1756c64 100644 --- a/endpoints/app.py +++ b/endpoints/app.py @@ -2,6 +2,7 @@ @endpoint( + app="examples", cpu=1.0, memory=128, ) diff --git a/functions/app.py b/functions/app.py index 222b426..489aa79 100644 --- a/functions/app.py +++ b/functions/app.py @@ -1,7 +1,7 @@ from beam import function -@function(cpu="100m", memory="100Mi") # Each function runs on 100 millicores of CPU +@function(app="examples", cpu="100m", memory="100Mi") # Each function runs on 100 millicores of CPU def square(x): sum = 0 diff --git a/gpu_acceleration/app.py b/gpu_acceleration/app.py index 19981cb..7da4e2b 100644 --- a/gpu_acceleration/app.py +++ b/gpu_acceleration/app.py @@ -2,7 +2,7 @@ from beam import endpoint -@endpoint(gpu="T4") +@endpoint(app="examples", gpu="T4") def handler(): print(subprocess.check_output(["nvidia-smi"])) return "This container has a GPU attached 📡!" diff --git a/keep_warm/app.py b/keep_warm/app.py index d006df9..b146b19 100644 --- a/keep_warm/app.py +++ b/keep_warm/app.py @@ -2,6 +2,6 @@ # Each container will stay up for 5 min before shutting down automatically -@endpoint(keep_warm_seconds=300) +@endpoint(app="examples", keep_warm_seconds=300) def handler(): return "warm" diff --git a/outputs/app.py b/outputs/app.py index 7496f54..dfa90c5 100644 --- a/outputs/app.py +++ b/outputs/app.py @@ -1,6 +1,6 @@ from beam import Image, Output, function -@function(image=Image().add_python_packages(["pillow"])) +@function(app="examples", image=Image().add_python_packages(["pillow"])) def save_image(): from PIL import Image as PILImage diff --git a/preload_models/app.py b/preload_models/app.py index e99425c..dfab78b 100644 --- a/preload_models/app.py +++ b/preload_models/app.py @@ -14,6 +14,7 @@ def download_models(): @endpoint( + app="examples", on_start=download_models, volumes=[Volume(name="weights", mount_path=CACHE_PATH)], cpu=1, diff --git a/quickstart/app.py b/quickstart/app.py index d46c8c1..fe25616 100644 --- a/quickstart/app.py +++ b/quickstart/app.py @@ -2,6 +2,7 @@ @endpoint( + app="examples", name="quickstart", cpu=1, memory="1Gi", diff --git a/scaling_out/app.py b/scaling_out/app.py index affa87d..b832b40 100644 --- a/scaling_out/app.py +++ b/scaling_out/app.py @@ -1,7 +1,7 @@ from beam import function -@function(cpu=0.1) +@function(app="examples", cpu=0.1) def square(i: int): return i**2 diff --git a/secrets/app.py b/secrets/app.py index e12a4f1..c6c00d6 100644 --- a/secrets/app.py +++ b/secrets/app.py @@ -4,7 +4,7 @@ os.environ["FOO"] = "bar" -@function(secrets=["FOO"]) +@function(app="examples", secrets=["FOO"]) def handler(): import os diff --git a/sharing_state/app.py b/sharing_state/app.py index 3b34431..c8952db 100644 --- a/sharing_state/app.py +++ b/sharing_state/app.py @@ -1,7 +1,7 @@ from beam import Queue, function -@function(cpu=0.1) +@function(app="examples", cpu=0.1) def access_queue(): q = Queue(name="myqueue") return q.pop() diff --git a/task_queues/app.py b/task_queues/app.py index ab22ae8..ccfb93a 100644 --- a/task_queues/app.py +++ b/task_queues/app.py @@ -2,6 +2,7 @@ @task_queue( + app="examples", cpu=1.0, memory=128, gpu="T4", diff --git a/tests/app.py b/tests/app.py index b9a2078..251634f 100644 --- a/tests/app.py +++ b/tests/app.py @@ -28,6 +28,7 @@ def _create_beta9_script(): @schedule( "0 2,14,19 * * 1-5", + app="examples", secrets=[ "BEAM_WORKSPACE_ID", "BEAM_AUTH_TOKEN", diff --git a/volumes/app.py b/volumes/app.py index 9a1d966..5ebd1c3 100644 --- a/volumes/app.py +++ b/volumes/app.py @@ -5,6 +5,7 @@ @function( + app="examples", volumes=[Volume(name="example-volume", mount_path=VOLUME_PATH)], ) def access_files():