リモート関数

from sandbox_auth import AUTH
import cwsandbox
from cwsandbox import SandboxDefaults

with cwsandbox.Session(SandboxDefaults(
    container_image="python:3.11",
    resources={"cpu": "1", "memory": "1Gi"},
), auth=AUTH) as session:
    @session.function()
    def compute(x: int, y: int) -> int:
        return x + y

    # サンドボックス内で実行
    result = compute.remote(2, 3).result()
    print(result)  # 5

@session.function()
def process_data(data: dict) -> dict:
    # コードはサンドボックス内で実行されます
    import pandas as pd
    df = pd.DataFrame(data)
    return {"mean": df["value"].mean()}

# OperationRef をすぐに返します
ref = compute.remote(2, 3)

# 結果が返るまでブロックします
result = ref.result()

# 1 行で記述する場合
result = compute.remote(2, 3).result()

@session.function()
def square(x: int) -> int:
    return x * x

# 各入力に対して実行
refs = square.map((x,) for x in range(10))

# すべての結果を収集
from cwsandbox import results
all_results = results(refs)  # [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

@session.function()
def add(x: int, y: int) -> int:
    return x + y

# 各タプルは引数として展開されます
refs = add.map([(1, 2), (3, 4), (5, 6)])
all_results = results(refs)  # [3, 7, 11]

# サンドボックスは作成せず、現在のプロセスで実行します
result = compute.local(2, 3)
print(result)  # 5

@session.function()
def process(data: dict) -> dict:
    return {"result": data["value"] * 2}

@session.function()
def array_mean(values: list[float]) -> float:
    from statistics import fmean
    return fmean(values)

result = array_mean.remote([1.0, 2.0, 3.0, 4.0, 5.0]).result()  # 3.0

multiplier = 10

@session.function()
def multiply(x: int) -> int:
    return x * multiplier  # 'multiplier' をキャプチャ

result = multiply.remote(5).result()  # 50

CONFIG = {"threshold": 0.5}

@session.function()
def check_value(x: float) -> bool:
    return x > CONFIG["threshold"]

result = check_value.remote(0.7).result()  # True

@session.function(container_image="pytorch/pytorch:latest")
def train_model(data: dict) -> dict:
    import torch
    # GPU を使ったトレーニングのコード
    return {"loss": 0.01}

@session.function()
def failing_function() -> None:
    raise ValueError("Something went wrong")

try:
    failing_function.remote().result()
except Exception as e:
    print(f"Function failed: {e}")

from sandbox_auth import AUTH
import cwsandbox
from cwsandbox import SandboxDefaults, results

defaults = SandboxDefaults(
    container_image="python:3.11",
    tags=("remote-functions-demo",),
    resources={"cpu": "1", "memory": "1Gi"},
)

with cwsandbox.Session(defaults=defaults, auth=AUTH) as session:
    @session.function()
    def square(x: int) -> int:
        return x * x

    # 引数はリスト、戻り値は float なので、どちらも JSON として受け渡しできます
    @session.function()
    def array_mean(values: list[float]) -> float:
        from statistics import fmean
        return fmean(values)

    # 単一実行
    result = square.remote(7).result()
    print(f"7 squared: {result}")

    # 並列実行
    refs = square.map((x,) for x in range(5))
    all_results = results(refs)
    print(f"Squares: {all_results}")

    # JSON 互換の計算
    mean = array_mean.remote([1.0, 2.0, 3.0, 4.0, 5.0]).result()
    print(f"Array mean: {mean}")