fix tests

python-sdk/indexify/executor/downloader.py

@@ -85,33 +85,31 @@ async def download_input(self, task: Task) -> DownloadedInputs:
             if response.headers["content-type"] == JsonSerializer.content_type
             else "cloudpickle"
         )
-        serializer = get_serializer(encoder)
-
         if task.invocation_id == input_id:
             return DownloadedInputs(
                 input=IndexifyData(
                     payload=response.content, id=input_id, encoder=encoder
                 ),
             )
 
-        deserialized_content = serializer.deserialize(response.content)
+        input_payload = response.content
 
         if reducer_url:
-            init_value = self._client.get(reducer_url)
+            response = self._client.get(reducer_url)
             try:
-                init_value.raise_for_status()
+                response.raise_for_status()
+                init_value = response.content
             except httpx.HTTPStatusError as e:
                 logger.error(
                     "failed to download reducer output",
                     url=reducer_url,
-                    error=init_value.text,
+                    error=response.text,
                 )
                 raise
-            init_value = serializer.deserialize(init_value.content)
             return DownloadedInputs(
                 input=IndexifyData(
                     input_id=task.invocation_id,
-                    payload=deserialized_content,
+                    payload=input_payload,
                     encoder=encoder,
                 ),
                 init_value=IndexifyData(
@@ -122,7 +120,7 @@ async def download_input(self, task: Task) -> DownloadedInputs:
         return DownloadedInputs(
             input=IndexifyData(
                 input_id=task.invocation_id,
-                payload=deserialized_content,
+                payload=input_payload,
                 encoder=encoder,
             )
         )

python-sdk/indexify/http_client.py

@@ -351,10 +351,12 @@ def _download_output(
         )
         response.raise_for_status()
         content_type = response.headers.get("Content-Type")
-        serializer = get_serializer(content_type)
-        decoded_response = serializer.deserialize(response.content)
+        if content_type == "application/octet-stream":
+            encoding = "cloudpickle"
+        else:
+            encoding = "json"
         return IndexifyData(
-            id=output_id, payload=decoded_response, encoder=serializer.encoding_type
+            id=output_id, payload=response.content, encoder=encoding
         )
 
     def graph_outputs(