From 84f610c0e9e612461f6b010db27a1b2169e21f11 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Sun, 9 Mar 2025 08:40:40 -0700 Subject: [PATCH] style: Fix linting issues in test_reasoning.py --- tests/basic/test_reasoning.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/basic/test_reasoning.py b/tests/basic/test_reasoning.py index b9b5363bd..b9db9c6c4 100644 --- a/tests/basic/test_reasoning.py +++ b/tests/basic/test_reasoning.py @@ -90,7 +90,9 @@ class TestReasoning(unittest.TestCase): # Mock streaming response chunks class MockStreamingChunk: - def __init__(self, content=None, reasoning_content=None, reasoning=None, finish_reason=None): + def __init__( + self, content=None, reasoning_content=None, reasoning=None, finish_reason=None + ): self.choices = [MagicMock()] self.choices[0].delta = MagicMock() self.choices[0].finish_reason = finish_reason @@ -108,7 +110,7 @@ class TestReasoning(unittest.TestCase): else: # Need to handle attribute access that would raise AttributeError delattr(self.choices[0].delta, "reasoning_content") - + # Set reasoning if provided if reasoning is not None: self.choices[0].delta.reasoning = reasoning @@ -271,7 +273,9 @@ class TestReasoning(unittest.TestCase): # Mock streaming response chunks class MockStreamingChunk: - def __init__(self, content=None, reasoning_content=None, reasoning=None, finish_reason=None): + def __init__( + self, content=None, reasoning_content=None, reasoning=None, finish_reason=None + ): self.choices = [MagicMock()] self.choices[0].delta = MagicMock() self.choices[0].finish_reason = finish_reason @@ -289,7 +293,7 @@ class TestReasoning(unittest.TestCase): else: # Need to handle attribute access that would raise AttributeError delattr(self.choices[0].delta, "reasoning_content") - + # Set reasoning if provided if reasoning is not None: self.choices[0].delta.reasoning = reasoning