full copy of metadata from private data to syn data

microsoft · Dec 27, 2024 · bbd6c54 · bbd6c54
1 parent aed239a
commit bbd6c54
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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.