From 7bd7d9f56ca6539f99e5df09fe797bdf6b4501f9 Mon Sep 17 00:00:00 2001
From: Felipe Mello <felipemello@fb.com>
Date: Mon, 10 Nov 2025 14:22:34 -0800
Subject: [PATCH] make method async

---
 apps/grpo/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/apps/grpo/main.py b/apps/grpo/main.py
index 10ea43d58..693dc8d81 100644
--- a/apps/grpo/main.py
+++ b/apps/grpo/main.py
@@ -210,7 +210,7 @@ class DatasetActor(ForgeActor):
     model: str = "Qwen/Qwen3-1.7B"
 
     @endpoint
-    def setup(self):
+    async def setup(self):
         self._tokenizer = get_tokenizer(self.model)
         self._epoch = 0