File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
src/data_designer/engine/column_generators/generators Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -115,9 +115,12 @@ def _sample_records(self, num_records: int) -> pd.DataFrame:
115115 logger .info (f" |-- seed dataset size: { self ._seed_dataset_size } records" )
116116 logger .info (f" |-- sampling strategy: { self .config .sampling_strategy } " )
117117 if self ._index_range is not None :
118- logger .info (
119- f" |-- selection strategy: { type (self .config .selection_strategy ).__name__ } \n { self .config .selection_strategy .model_dump_json (indent = 4 )} "
120- )
118+ if isinstance (self .config .selection_strategy , IndexRange ):
119+ logger .info (f" |-- selection: rows [{ self ._index_range .start } to { self ._index_range .end } ] inclusive" )
120+ else :
121+ logger .info (
122+ f" |-- selection: partition { self .config .selection_strategy .index + 1 } of { self .config .selection_strategy .num_partitions } "
123+ )
121124 logger .info (f" |-- seed dataset size after selection: { self ._index_range .size } records" )
122125 df_batch = pd .DataFrame ()
123126 df_sample = pd .DataFrame () if self ._df_remaining is None else self ._df_remaining
You can’t perform that action at this time.
0 commit comments