diff --git a/src/synthcity/plugins/privacy/plugin_decaf.py b/src/synthcity/plugins/privacy/plugin_decaf.py index bed70395..183ff9c3 100644 --- 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 index daca0370..9e338d28 100644 --- 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))