Skip to content

Commit f1c1ec8

Browse files
committed
addressing comment
1 parent 98f565e commit f1c1ec8

File tree

1 file changed

+11
-14
lines changed

1 file changed

+11
-14
lines changed

src/data_designer/config/utils/visualization.py

Lines changed: 11 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -104,24 +104,21 @@ def display_sample_record(
104104
except IndexError:
105105
raise DatasetSampleDisplayError(f"Index {i} is out of bounds for dataset of length {num_records}.")
106106

107+
processor_data_to_display = None
107108
if self._has_processor_artifact_preview() and len(self.processor_artifact_preview) > 0:
108109
if processors_to_display is None:
109110
processors_to_display = list(self.processor_artifact_preview.keys())
110111

111-
if len(processors_to_display) == 0:
112-
raise DatasetSampleDisplayError("No processors to display.")
113-
114-
processor_data_to_display = {}
115-
for processor in processors_to_display:
116-
if (
117-
isinstance(self.processor_artifact_preview[processor], list)
118-
and len(self.processor_artifact_preview[processor]) == num_records
119-
):
120-
processor_data_to_display[processor] = self.processor_artifact_preview[processor][i]
121-
else:
122-
processor_data_to_display[processor] = self.processor_artifact_preview[processor]
123-
else:
124-
processor_data_to_display = None
112+
if len(processors_to_display) > 0:
113+
processor_data_to_display = {}
114+
for processor in processors_to_display:
115+
if (
116+
isinstance(self.processor_artifact_preview[processor], list)
117+
and len(self.processor_artifact_preview[processor]) == num_records
118+
):
119+
processor_data_to_display[processor] = self.processor_artifact_preview[processor][i]
120+
else:
121+
processor_data_to_display[processor] = self.processor_artifact_preview[processor]
125122

126123
display_sample_record(
127124
record=record,

0 commit comments

Comments
 (0)