From 5b18e5fd428766a6e53781bbf14b35d17b2ebbab Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 11 Jan 2025 22:13:59 +0000
Subject: [PATCH] chore: lint

---
 tests/unit/test_sync.py | 28 +++++++++++++++++++---------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/tests/unit/test_sync.py b/tests/unit/test_sync.py
index 7b33a1be..d089fc81 100644
--- a/tests/unit/test_sync.py
+++ b/tests/unit/test_sync.py
@@ -485,7 +485,9 @@ def check_sync():
                 # confirm local and remote are synced
                 assert route_layer.is_synced()
                 # now confirm utterances are correct
-                local_utterances = route_layer.index.get_utterances(include_metadata=True)
+                local_utterances = route_layer.index.get_utterances(
+                    include_metadata=True
+                )
                 # we sort to ensure order is the same
                 # TODO JB: there is a bug here where if we include_metadata=True it fails
                 local_utterances.sort(key=lambda x: x.to_str(include_metadata=True))
@@ -534,11 +536,11 @@ def check_r1_utterances():
                     Utterance(name="Route 2", utterances="Goodbye"),
                     Utterance(name="Route 3", utterances="Boo"),
                 ]
-                local_utterances = route_layer.index.get_utterances(include_metadata=True)
-                # we sort to ensure order is the same
-                local_utterances.sort(
-                    key=lambda x: x.to_str(include_metadata=True)
+                local_utterances = route_layer.index.get_utterances(
+                    include_metadata=True
                 )
+                # we sort to ensure order is the same
+                local_utterances.sort(key=lambda x: x.to_str(include_metadata=True))
                 assert local_utterances == r1_utterances
 
             check_r1_utterances()
@@ -554,7 +556,9 @@ def check_r1_utterances():
             def check_r2_utterances():
                 # confirm local and remote are synced
                 assert route_layer.is_synced()
-                local_utterances = route_layer.index.get_utterances(include_metadata=True)
+                local_utterances = route_layer.index.get_utterances(
+                    include_metadata=True
+                )
                 # we sort to ensure order is the same
                 local_utterances.sort(key=lambda x: x.to_str(include_metadata=True))
                 assert local_utterances == [
@@ -619,7 +623,9 @@ def check_sync():
                 # confirm local and remote are synced
                 assert route_layer.is_synced()
                 # now confirm utterances are correct
-                local_utterances = route_layer.index.get_utterances(include_metadata=True)
+                local_utterances = route_layer.index.get_utterances(
+                    include_metadata=True
+                )
                 # we sort to ensure order is the same
                 local_utterances.sort(key=lambda x: x.to_str(include_metadata=True))
                 assert local_utterances == [
@@ -809,7 +815,9 @@ async def test_auto_sync_local(
 
             @async_retry(max_retries=RETRY_COUNT, delay=PINECONE_SLEEP)
             async def check_sync():
-                assert await route_layer.index.aget_utterances(include_metadata=True) == [
+                assert await route_layer.index.aget_utterances(
+                    include_metadata=True
+                ) == [
                     Utterance(route="Route 1", utterance="Hello"),
                     Utterance(route="Route 2", utterance="Hi"),
                 ], "The routes in the index should match the local routes"
@@ -842,7 +850,9 @@ async def test_auto_sync_remote(
 
             @async_retry(max_retries=RETRY_COUNT, delay=PINECONE_SLEEP)
             async def check_sync():
-                assert await route_layer.index.aget_utterances(include_metadata=True) == [
+                assert await route_layer.index.aget_utterances(
+                    include_metadata=True
+                ) == [
                     Utterance(route="Route 1", utterance="Hello"),
                     Utterance(route="Route 2", utterance="Hi"),
                 ], "The routes in the index should match the local routes"