From bbd6c5401dbb428b82431598bd0a7ad1ee470712 Mon Sep 17 00:00:00 2001
From: Zinan Lin <linzinan1995@126.com>
Date: Fri, 27 Dec 2024 15:29:30 -0800
Subject: [PATCH] full copy of metadata from private data to syn data

---
 pe/runner/pe.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pe/runner/pe.py b/pe/runner/pe.py
index cacbc10..aaa97ec 100644
--- a/pe/runner/pe.py
+++ b/pe/runner/pe.py
@@ -1,4 +1,5 @@
 import numpy as np
+import copy
 
 from pe.dp import Gaussian
 from pe.data import Data
@@ -171,8 +172,8 @@ def run(
                     syn_data_list.append(syn_data)
                 syn_data = Data.concat(syn_data_list, metadata=self._priv_data.metadata)
                 syn_data.data_frame.reset_index(drop=True, inplace=True)
+                syn_data.metadata = copy.deepcopy(self._priv_data.metadata)
                 syn_data.metadata.iteration = 0
-                syn_data.metadata.label_info = self._priv_data.metadata.label_info
                 self._log_metrics(syn_data)
 
             # Run PE iterations.