Skip to content

Commit cfd1911

Browse files
committed
large mct batch list check fix
1 parent 2ac5ecc commit cfd1911

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

pat2vec/patvec_get_batch_methods/get_merged_batches.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -571,6 +571,11 @@ def get_merged_pat_batch_mct_docs(
571571
# Apply data type filters for MCT documents
572572
batch_target = apply_data_type_mct_docs_filters(config_obj, batch_target)
573573

574+
# larger batches returned as lists...
575+
if isinstance(batch_target, list) and len(batch_target) == 1:
576+
577+
batch_target = batch_target[0]
578+
574579
# Drop rows with NaN values in critical columns
575580
col_list_drop_nan = [
576581
"observation_valuetext_analysed",

0 commit comments

Comments
 (0)