chore: linting

airtai · Jun 20, 2024 · 7b92dd1 · 7b92dd1
1 parent 7bd6151
commit 7b92dd1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tests/brokers/nats/test_test_client.py b/tests/brokers/nats/test_test_client.py
@@ -91,7 +91,6 @@ async def test_inbox_prefix_with_real(
             assert br._connection._inbox_prefix == b"test"
             assert "test" in str(br._connection.new_inbox())
 
-
     async def test_respect_middleware(self, queue):
         routes = []
 

diff --git a/tests/tools.py b/tests/tools.py
@@ -7,10 +7,12 @@ def spy_decorator(method):
     mock = MagicMock()
 
     if inspect.iscoroutinefunction(method):
+
         async def wrapper(*args, **kwargs):
             mock(*args, **kwargs)
             return await method(*args, **kwargs)
     else:
+
         def wrapper(*args, **kwargs):
             mock(*args, **kwargs)
             return method(*args, **kwargs)