リモート関数

このチュートリアルでは、リモート関数を使用してサンドボックス内で Python 関数を実行する方法を説明します。シェルコマンドの文字列を組み立てる代わりに、サンドボックスで実行されるコードをローカル関数のように呼び出しできます。このチュートリアルは、Python コードからサンドボックス内のコンピュートを直接呼び出し、型付きの結果を取得したい開発者を対象としています。

認証を選択する

セクションで CKS が指定されていない限り、これらの例はサーバーレスキャパシティ上で実行されます。サンドボックスを所有するアカウントの認証情報を選択してください。キーまたはトークンの取得方法については、認証情報を選択して取得するを参照してください。インストールコマンドは、有効化した Python 仮想環境内で実行してください。使用する認証情報のタブにある Python コードを、このページで実行する Python スクリプトと同じディレクトリに sandbox_auth.py として保存します。このファイルに保存されるのはどの認証方式を使うかという設定であり、認証情報そのものではありません。例ではこのファイルから AUTH をインポートし、サンドボックスまたはセッションを新規作成するたびに明示的に渡します。

Forge API キー
CoreWeave API アクセストークン

SDK の W&B 認証方式で Forge APIキーを使用します。

uv pip install 'cwsandbox[wandb]>=1.14.2'
export WANDB_API_KEY="[WANDB-API-KEY]"

sandbox_auth.py

from cwsandbox import AuthStrategy

AUTH = AuthStrategy.WANDB

CoreWeave 組織のトークンを使用します。

uv pip install 'cwsandbox>=1.14.2'
export CWSANDBOX_API_KEY="[API-ACCESS-TOKEN]"

sandbox_auth.py

from cwsandbox import AuthStrategy

AUTH = AuthStrategy.COREWEAVE_API_KEY

CKS に配置する場合は、CoreWeave トークンを選択し、ランナーを有効化したうえで、サンドボックスの作成時または SandboxDefaults で placement_mode="cks" を設定します。auth=AUTH を指定してセッションを作成すると、そのセッションが所有するサンドボックスも認証されます。サンドボックスへの再接続、一覧表示、クリーンアップを行う際も、同じ認証情報を使用してください。

リモート関数を定義して呼び出しする

関数を @session.function() でデコレートすると、.remote() を使ってサンドボックス内でその関数を呼び出しできます。呼び出しは参照を返すので、.result() で解決します:

from sandbox_auth import AUTH
from cwsandbox import Sandbox, SandboxDefaults

with Sandbox.session(SandboxDefaults(
    resources={"cpu": "1", "memory": "1Gi"},
), auth=AUTH) as session:
    @session.function()
    def add(x: int, y: int) -> int:
        return x + y

    result = add.remote(2, 3).result()  # 5

`.map()` で関数を並列に実行する

.map() を使用すると、多数の invocation を一度にディスパッチして、その結果を収集できます。独立した Call は 1 つずつではなく同時に実行されます。

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

refs = square.map([(1,), (2,), (3,)])
results = [r.result() for r in refs]  # [1, 4, 9]

次のステップ

これで、サンドボックス内の Python 関数をローカル関数と同じように呼び出せるようになり、.map() を使えば処理を並列に展開できます。シリアライズモードや .local() によるテストなどの詳細については、リモート関数ガイドを参照してください。

サンドボックス

サンドボックスを使用する

サンドボックスの管理

リファレンス

認証を選択する

リモート関数を定義して呼び出しする

`.map()` で関数を並列に実行する

次のステップ

​認証を選択する

​リモート関数を定義して呼び出しする

​.map() で関数を並列に実行する

​次のステップ

認証を選択する

リモート関数を定義して呼び出しする

`.map()` で関数を並列に実行する

次のステップ