Skip to content

Commit 7bfaf51

Browse files
committed
fixup! fix failing tests
1 parent f0a471b commit 7bfaf51

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

Framework/Core/src/ArrowTableSlicingCache.cxx

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -97,11 +97,11 @@ void ArrowTableSlicingCache::setCaches(Cache&& bsks, Cache&& bsksUnsorted)
9797

9898
arrow::Status ArrowTableSlicingCache::updateCacheEntry(int pos, std::shared_ptr<arrow::Table> const& table)
9999
{
100+
values[pos].reset();
101+
counts[pos].reset();
102+
offsets[pos].clear();
103+
sizes[pos].clear();
100104
if (table->num_rows() == 0) {
101-
values[pos].reset();
102-
counts[pos].reset();
103-
offsets[pos].clear();
104-
sizes[pos].clear();
105105
return arrow::Status::OK();
106106
}
107107
auto& [b, k, e] = bindingsKeys[pos];
@@ -137,11 +137,12 @@ arrow::Status ArrowTableSlicingCache::updateCacheEntry(int pos, std::shared_ptr<
137137
int64_t offset = 0;
138138
for (auto i = 0U; i < values[pos]->length(); ++i) {
139139
auto value = values[pos]->Value(i);
140+
auto count = counts[pos]->Value(i);
140141
if (value >= 0) {
141142
offsets[pos][value] = offset;
142-
sizes[pos][value] = counts[pos]->Value(i);
143+
sizes[pos][value] = count;
143144
}
144-
offset += counts[pos]->Value(i);
145+
offset += count;
145146
}
146147
return arrow::Status::OK();
147148
}

0 commit comments

Comments
 (0)