From b0ad30cded8091dfdf2d9c505bb4598fdf7879ab Mon Sep 17 00:00:00 2001
From: Diptanu Gon Choudhury <diptanuc@gmail.com>
Date: Mon, 2 Dec 2024 15:32:20 -0800
Subject: [PATCH] lint

---
 python-sdk/indexify/executor/agent.py         |  2 +-
 python-sdk/indexify/executor/task_reporter.py |  3 ++-
 python-sdk/indexify/functions_sdk/graph.py    |  2 +-
 python-sdk/indexify/http_client.py            |  4 +---
 python-sdk/tests/test_graph_behaviours.py     | 16 +++++++++-------
 5 files changed, 14 insertions(+), 13 deletions(-)

diff --git a/python-sdk/indexify/executor/agent.py b/python-sdk/indexify/executor/agent.py
index bdbc5c529..4d597bc1f 100644
--- a/python-sdk/indexify/executor/agent.py
+++ b/python-sdk/indexify/executor/agent.py
@@ -5,9 +5,9 @@
 from pathlib import Path
 from typing import Dict, List, Optional
 
+import structlog
 from httpx_sse import aconnect_sse
 from pydantic import BaseModel
-import structlog
 
 from indexify.common_util import get_httpx_client
 from indexify.functions_sdk.data_objects import (
diff --git a/python-sdk/indexify/executor/task_reporter.py b/python-sdk/indexify/executor/task_reporter.py
index 2426a721b..6d160c360 100644
--- a/python-sdk/indexify/executor/task_reporter.py
+++ b/python-sdk/indexify/executor/task_reporter.py
@@ -1,9 +1,10 @@
 from typing import Optional
-import structlog
 
 import nanoid
+import structlog
 from httpx import Timeout
 from pydantic import BaseModel
+
 from indexify.common_util import get_httpx_client
 from indexify.executor.api_objects import RouterOutput as ApiRouterOutput
 from indexify.executor.api_objects import TaskResult
diff --git a/python-sdk/indexify/functions_sdk/graph.py b/python-sdk/indexify/functions_sdk/graph.py
index 375bb259d..be0b3b688 100644
--- a/python-sdk/indexify/functions_sdk/graph.py
+++ b/python-sdk/indexify/functions_sdk/graph.py
@@ -1,5 +1,5 @@
-import sys
 import json
+import sys
 from collections import defaultdict
 from queue import deque
 from typing import (
diff --git a/python-sdk/indexify/http_client.py b/python-sdk/indexify/http_client.py
index 98ba0d520..7a6a9d3d2 100644
--- a/python-sdk/indexify/http_client.py
+++ b/python-sdk/indexify/http_client.py
@@ -355,9 +355,7 @@ def _download_output(
             encoding = "cloudpickle"
         else:
             encoding = "json"
-        return IndexifyData(
-            id=output_id, payload=response.content, encoder=encoding
-        )
+        return IndexifyData(id=output_id, payload=response.content, encoder=encoding)
 
     def graph_outputs(
         self,
diff --git a/python-sdk/tests/test_graph_behaviours.py b/python-sdk/tests/test_graph_behaviours.py
index c1aa7169e..a1bfdc7f6 100644
--- a/python-sdk/tests/test_graph_behaviours.py
+++ b/python-sdk/tests/test_graph_behaviours.py
@@ -1,11 +1,10 @@
 import unittest
 from pathlib import Path
 from typing import List, Union
-from typing_extensions import TypedDict
-
 
 from parameterized import parameterized
 from pydantic import BaseModel
+from typing_extensions import TypedDict
 
 from indexify import (
     Graph,
@@ -120,8 +119,11 @@ def sum_of_squares(init_value: Sum, x: int) -> Sum:
     init_value.val += x
     return init_value
 
+
 class JsonSum(TypedDict):
     val: int
+
+
 @indexify_function(accumulate=JsonSum, encoder="json")
 def sum_of_squares_with_json_encoding(init_value: JsonSum, x: int) -> JsonSum:
     val = init_value.get("val", 0)
@@ -391,11 +393,11 @@ def test_invoke_file(self, is_remote):
 
     @parameterized.expand([(False), (True)])
     def test_pipeline(self, is_remote):
-       p = create_simple_pipeline()
-       p = remote_or_local_pipeline(p, is_remote)
-       invocation_id = p.run(block_until_done=True, x=3)
-       output = p.output(invocation_id, "make_it_string")
-       self.assertEqual(output, ["5"])
+        p = create_simple_pipeline()
+        p = remote_or_local_pipeline(p, is_remote)
+        invocation_id = p.run(block_until_done=True, x=3)
+        output = p.output(invocation_id, "make_it_string")
+        self.assertEqual(output, ["5"])
 
     @parameterized.expand([(False), (True)])
     def test_ignore_none_in_map(self, is_remote):