Skip to content

Commit 4e3bd3e

Browse files
committed
update log message
1 parent 80fca51 commit 4e3bd3e

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/data_designer/engine/column_generators/generators/seed_dataset.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)