Skip to content

Commit 0691277

Browse files
committed
try #4
1 parent 37f1818 commit 0691277

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

  • sdk/python/feast/infra/compute_engines/local

sdk/python/feast/infra/compute_engines/local/nodes.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -263,9 +263,14 @@ def execute(self, context: ExecutionContext) -> ArrowTableValue:
263263
if input_table.num_rows == 0:
264264
return input_table
265265

266-
chunk_size = getattr(
266+
raw_chunk_size = getattr(
267267
getattr(context.repo_config, "batch_engine", None), "chunk_size", None
268268
)
269+
try:
270+
chunk_size = int(raw_chunk_size) if raw_chunk_size is not None else None
271+
except (TypeError, ValueError):
272+
chunk_size = None
273+
269274
if chunk_size and chunk_size > 0:
270275
batches = input_table.to_batches(max_chunksize=chunk_size)
271276
else:

0 commit comments

Comments
 (0)