Skip to content

Commit

Permalink
[GitHub Action] Lint and format code with Ruff
Browse files Browse the repository at this point in the history
  • Loading branch information
actions-user committed Nov 25, 2024
1 parent 06c3f12 commit b76495c
Showing 1 changed file with 18 additions and 6 deletions.
24 changes: 18 additions & 6 deletions aifn/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -735,12 +735,24 @@ def batch_query(
loop = asyncio.get_running_loop()
if loop.is_running():
nest_asyncio.apply()
task = loop.create_task(self._batch_query(
fn_name=fn_name, version=version, batch_inputs=batch_inputs, return_reasoning=return_reasoning, strict=strict
))
task = loop.create_task(
self._batch_query(
fn_name=fn_name,
version=version,
batch_inputs=batch_inputs,
return_reasoning=return_reasoning,
strict=strict,
)
)
return loop.run_until_complete(task)
except RuntimeError:
# If no event loop is running, use asyncio.run
return asyncio.run(self._batch_query(
fn_name=fn_name, version=version, batch_inputs=batch_inputs, return_reasoning=return_reasoning, strict=strict
))
return asyncio.run(
self._batch_query(
fn_name=fn_name,
version=version,
batch_inputs=batch_inputs,
return_reasoning=return_reasoning,
strict=strict,
)
)

0 comments on commit b76495c

Please sign in to comment.