Skip to content

Commit 86cd890

Browse files
authored
Add support for async method to be used with evaluate api (Azure#33109)
1 parent 10dde0f commit 86cd890

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

sdk/ai/azure-ai-generative/azure/ai/generative/evaluate/_local_code_handler.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,16 @@ def generate_prediction_data(self):
2727
prediction_data = []
2828
test_data = self.get_test_data_as_jsonl()
2929

30+
import inspect
31+
is_asset_async = False
32+
if inspect.iscoroutinefunction(self.asset):
33+
is_asset_async = True
34+
import asyncio
35+
3036
for d in test_data:
31-
prediction_data.append(self.asset(**d))
37+
prediction_data.append(
38+
asyncio.run(self.asset(**d)) if is_asset_async else self.asset(**d)
39+
)
40+
3241

3342
return prediction_data

0 commit comments

Comments
 (0)