Skip to content

Commit 8068e49

Browse files
author
Tyler Payne
committed
Fix formatting errors
1 parent 9dfd8ac commit 8068e49

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

python/packages/autogen-ext/tests/tools/test_mcp_actor.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -469,7 +469,9 @@ async def test_run_actor_exception_handling() -> None:
469469

470470
# Create a mock session that will raise exceptions by modifying our mock session
471471
@asynccontextmanager
472-
async def mock_failing_session(server_params: Any, sampling_callback: Any = None) -> AsyncGenerator[MagicMock, None]:
472+
async def mock_failing_session(
473+
server_params: Any, sampling_callback: Any = None
474+
) -> AsyncGenerator[MagicMock, None]:
473475
mock_session = MagicMock()
474476
mock_session.initialize = AsyncMock(
475477
return_value=mcp_types.InitializeResult(
@@ -1140,7 +1142,9 @@ async def test_run_actor_all_command_types_exception_handling() -> None:
11401142

11411143
# Create a mock session that will raise exceptions for all command types
11421144
@asynccontextmanager
1143-
async def mock_failing_session(server_params: Any, sampling_callback: Any = None) -> AsyncGenerator[MagicMock, None]:
1145+
async def mock_failing_session(
1146+
server_params: Any, sampling_callback: Any = None
1147+
) -> AsyncGenerator[MagicMock, None]:
11441148
mock_session = MagicMock()
11451149
mock_session.initialize = AsyncMock(
11461150
return_value=mcp_types.InitializeResult(

0 commit comments

Comments
 (0)