clean up

vanderschaarlab · Sep 3, 2024 · e2acb6a · e2acb6a
1 parent 263dc8a
commit e2acb6a
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/src/synthcity/plugins/privacy/plugin_decaf.py b/src/synthcity/plugins/privacy/plugin_decaf.py
@@ -446,14 +446,12 @@ def _generate(
         **kwargs: Any,
     ) -> pd.DataFrame:
         encoded_biased_edges = self._encode_edges(biased_edges)
-        print(f"encoded_biased_edges = {encoded_biased_edges}")
 
         def _sample(count: int) -> pd.DataFrame:
             # generate baseline values
             seed_values = self.baseline_generator(count)
             seed_values = torch.from_numpy(seed_values).to(DEVICE)
             # debias baseline values
-            print("generating synthetic data")
             vals = (
                 self.model.gen_synthetic(seed_values, biased_edges=encoded_biased_edges)
                 .detach()

diff --git a/tests/plugins/privacy/test_dpgan.py b/tests/plugins/privacy/test_dpgan.py
@@ -48,7 +48,6 @@ def test_plugin_hyperparams(test_plugin: Plugin) -> None:
     "test_plugin", generate_fixtures(plugin_name, plugin, plugin_args)
 )
 def test_plugin_fit(test_plugin: Plugin) -> None:
-    print(test_plugin)
     X = pd.DataFrame(load_iris()["data"])
     test_plugin.fit(GenericDataLoader(X))