feat: Add method to configure DataFrame display options in PySessionConfig

kosiew · kosiew · commit b401e1a1a0f2 · 2025-04-02T15:03:06.000+08:00
diff --git a/src/context.rs b/src/context.rs
@@ -214,6 +214,12 @@ impl PySessionConfig {
         Self::from(self.config.clone().with_repartition_file_min_size(size))
     }
 
+    fn with_dataframe_display_config(&self, display_config: DataframeDisplayConfig) -> Self {
+        let mut config = self.clone();
+        config.display_config = display_config;
+        config
+    }
+
     fn with_parquet_pruning(&self, enabled: bool) -> Self {
         Self::from(self.config.clone().with_parquet_pruning(enabled))
     }

Original file line number	Diff line number	Diff line change
`@@ -214,6 +214,12 @@ impl PySessionConfig {`
`214`	`214`	`Self::from(self.config.clone().with_repartition_file_min_size(size))`
`215`	`215`	`}`
`216`	`216`
	`217`	`+ fn with_dataframe_display_config(&self, display_config: DataframeDisplayConfig) -> Self {`
	`218`	`+ let mut config = self.clone();`
	`219`	`+ config.display_config = display_config;`
	`220`	`+ config`
	`221`	`+ }`
	`222`	`+`
`217`	`223`	`fn with_parquet_pruning(&self, enabled: bool) -> Self {`
`218`	`224`	`Self::from(self.config.clone().with_parquet_pruning(enabled))`
`219`	`225`	`}`