From 96cca844d0023dc7ef4116d41d414d45d7448c71 Mon Sep 17 00:00:00 2001
From: Andrew Lamb <andrew@nerdnetworks.org>
Date: Fri, 14 Apr 2023 15:12:14 -0400
Subject: [PATCH] Add partition_aware to UnionExec display

---
 datafusion/core/src/physical_plan/union.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/datafusion/core/src/physical_plan/union.rs b/datafusion/core/src/physical_plan/union.rs
index 3d4522272a521..66bccf55063a8 100644
--- a/datafusion/core/src/physical_plan/union.rs
+++ b/datafusion/core/src/physical_plan/union.rs
@@ -318,7 +318,11 @@ impl ExecutionPlan for UnionExec {
     ) -> std::fmt::Result {
         match t {
             DisplayFormatType::Default => {
-                write!(f, "UnionExec")
+                write!(f, "UnionExec")?;
+                if self.partition_aware {
+                    write!(f, ": partition_aware=true")?;
+                }
+                Ok(())
             }
         }
     }