diff --git a/nodestream/pipeline/extractors/stores/aws/s3_extractor.py b/nodestream/pipeline/extractors/stores/aws/s3_extractor.py index 302fd1d81..803b1f413 100644 --- a/nodestream/pipeline/extractors/stores/aws/s3_extractor.py +++ b/nodestream/pipeline/extractors/stores/aws/s3_extractor.py @@ -1,5 +1,5 @@ from contextlib import asynccontextmanager -from io import IOBase +from io import BytesIO, IOBase from logging import getLogger from pathlib import Path from typing import Iterable, Optional @@ -47,7 +47,7 @@ async def as_reader(self, reader: IOBase): streaming_body = self.s3_client.get_object(Bucket=self.bucket, Key=self.key)[ "Body" ] - yield reader(streaming_body._raw_stream) + yield reader(BytesIO(streaming_body.read())) self.archive_if_required(self.key) diff --git a/pyproject.toml b/pyproject.toml index 10fef9891..a8680b8d9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "nodestream" -version = "0.12.5a11" +version = "0.12.5a13" description = "A Fast, Declarative ETL for Graph Databases." license = "GPL-3.0-only" authors = [