diff --git a/src/datachain/data_storage/warehouse.py b/src/datachain/data_storage/warehouse.py index 843095485..32dfb5385 100644 --- a/src/datachain/data_storage/warehouse.py +++ b/src/datachain/data_storage/warehouse.py @@ -219,7 +219,7 @@ def dataset_select_paginated( num_yielded = 0 # Ensure we're using a thread-local connection - with self.clone(use_new_connection=True) as wh: + with self.clone() as wh: while True: if limit is not None: limit -= num_yielded