Remove AIMGDP from dpsynth

bin/main.py

@@ -32,7 +32,6 @@
 import numpy as np
 import pandas as pd
 
-
 _DATASET_PATH = flags.DEFINE_string(
     'dataset',
     None,
@@ -63,7 +62,7 @@
 _MECHANISM = flags.DEFINE_enum(
     'mechanism',
     'mst',
-    ['mst', 'aim', 'independent', 'aim_gdp'],
+    ['mst', 'aim', 'independent'],
     'Mechanism to use.',
 )
 
@@ -100,8 +99,6 @@ def main(_):
       mechanism_config = dpsynth.discrete_mechanisms.AIMConfig()
     case 'independent':
       mechanism_config = dpsynth.discrete_mechanisms.IndependentConfig()
-    case 'aim_gdp':
-      mechanism_config = dpsynth.discrete_mechanisms.AIMGDPConfig()
     case _:
       raise ValueError(f'Unknown mechanism: {_MECHANISM.value}')
 

docs/api_reference.rst

@@ -133,7 +133,6 @@ marginal measurement and domain compression.
    IndependentConfig
    DirectConfig
    SWIFTConfig
-   AIMGDPConfig
 
 DiscreteConfig and DiscreteMechanism
 ------------------------------------

docs/in_memory_api.md

@@ -162,8 +162,7 @@ python3 bin/main.py \
     arguments via `--read_csv_args`).
 *   `--domain`: Path to the YAML domain specification file.
 *   `--epsilon`, `--delta`: Total DP privacy budget.
-*   `--mechanism`: Supported options are `mst`, `aim`, `independent`, and
-    `aim_gdp`.
+*   `--mechanism`: Supported options are `mst`, `aim`, and `independent`.
 *   `--seed`: Integer seed for reproducible randomness across DP sampling and
     PGM inference.
 *   `--output_path`: Destination filepath where the synthetic CSV will be

dpsynth/discrete_mechanisms/README.md

@@ -88,18 +88,6 @@ workload; `_allocate_budget()` reserves rho for the adaptive loop; `_run()`
 replaces the standard base execution path. Helper functions filter valid
 candidates and privately choose the worst-approximated marginal.
 
-## `aim_gdp.py` — AIM with GDP-Oriented Allocation
-
-Implements `AIMGDPConfig`, a variant of AIM with the same adaptive workflow
-but GDP units for its internal loop budgeting. It is useful when its alternative
-privacy-accounting behavior is preferred.
-
-**Public API:** `AIMGDPConfig(workload=...)`
-
-**Internal behavior:** Like `aim.py`, it overrides `_one_way_cliques()`,
-`_allocate_budget()`, and `_run()`. Its internal helpers compute GDP-aware error
-scores and select the next workload marginal.
-
 ## `swift.py` — Workload and Clique-Tree Mechanism
 
 Implements `SWIFT`, a workload-informed mechanism that selects

dpsynth/discrete_mechanisms/__init__.py

@@ -26,8 +26,6 @@
 from dpsynth.api import MechanismConfig
 from dpsynth.discrete_mechanisms.aim import AIM
 from dpsynth.discrete_mechanisms.aim import AIMConfig
-from dpsynth.discrete_mechanisms.aim_gdp import AIMGDP
-from dpsynth.discrete_mechanisms.aim_gdp import AIMGDPConfig
 from dpsynth.discrete_mechanisms.common import DiscreteMechanismResult
 from dpsynth.discrete_mechanisms.common import MechanismDiagnostics
 from dpsynth.discrete_mechanisms.direct import Direct
@@ -43,7 +41,6 @@
 
 # Backwards-compatible aliases.
 AIMMechanism = AIMConfig
-AIMGDPMechanism = AIMGDPConfig
 DirectMechanism = DirectConfig
 IndependentMechanism = IndependentConfig
 MSTMechanism = MSTConfig