Skip to content

Commit 984b906

Browse files
committed
fix: Add validation for display configuration properties in DataframeDisplayConfig
1 parent 354ff45 commit 984b906

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

python/datafusion/context.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,8 @@ def max_table_bytes(self) -> int:
119119
@max_table_bytes.setter
120120
def max_table_bytes(self, value: int) -> None:
121121
"""Set the maximum bytes to display for table presentation."""
122+
if value <= 0:
123+
raise ValueError("max_table_bytes must be greater than 0")
122124
self.config_internal.max_table_bytes = value
123125

124126
@property
@@ -129,6 +131,8 @@ def min_table_rows(self) -> int:
129131
@min_table_rows.setter
130132
def min_table_rows(self, value: int) -> None:
131133
"""Set the minimum number of table rows to display."""
134+
if value <= 0:
135+
raise ValueError("min_table_rows must be greater than 0")
132136
self.config_internal.min_table_rows = value
133137

134138
@property
@@ -139,6 +143,8 @@ def max_cell_length(self) -> int:
139143
@max_cell_length.setter
140144
def max_cell_length(self, value: int) -> None:
141145
"""Set the maximum length of a cell before it gets minimized."""
146+
if value <= 0:
147+
raise ValueError("max_cell_length must be greater than 0")
142148
self.config_internal.max_cell_length = value
143149

144150
@property
@@ -149,6 +155,8 @@ def max_table_rows_in_repr(self) -> int:
149155
@max_table_rows_in_repr.setter
150156
def max_table_rows_in_repr(self, value: int) -> None:
151157
"""Set the maximum number of rows to display in repr string output."""
158+
if value <= 0:
159+
raise ValueError("max_table_rows_in_repr must be greater than 0")
152160
self.config_internal.max_table_rows_in_repr = value
153161

154162

0 commit comments

Comments
 (0)