Skip to content

Commit

Permalink
Partially revert #532 (#535)
Browse files Browse the repository at this point in the history
  • Loading branch information
gtopper authored Sep 10, 2024
1 parent d36945a commit 43e8f4b
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
aiohttp~=3.8
v3io~=0.6.9
v3io~=0.6.10
# exclude pandas 1.5.0 due to https://github.com/pandas-dev/pandas/issues/48767
# and 1.5.* due to https://github.com/pandas-dev/pandas/issues/49203
# pandas 2.2 requires sqlalchemy 2
Expand Down
4 changes: 3 additions & 1 deletion storey/drivers.py
Original file line number Diff line number Diff line change
Expand Up @@ -532,7 +532,9 @@ def _build_simplified_feature_store_request(self, aggregation_element):

if use_parallel:
for attr_name, d in pexpressions.items():
encoded_array = kv_array.encode_list(d["values"][d["first_index"] : d["last_index"] + 1])
encoded_array = kv_array.encode_list(d["values"][d["first_index"] : d["last_index"] + 1]).decode(
"utf-8"
)
paggregate = self.parallel_aggregates[d["aggregation"]]
sliced_array = f'{attr_name}[{d["first_index"]}..{d["last_index"]}]'
expressions.append(f"{sliced_array}={paggregate}({sliced_array}, blob('{encoded_array}'))")
Expand Down

0 comments on commit 43e8f4b

Please sign in to comment.