fix basic inputs for distribution reconstruction

qiskit_addon_cutting/__init__.py

@@ -88,7 +88,10 @@
     PartitionedCuttingProblem,
 )
 from .cutting_experiments import generate_cutting_experiments
-from .cutting_reconstruction import reconstruct_expectation_values
+from .cutting_reconstruction import (
+    reconstruct_expectation_values,
+    reconstruct_distribution,
+)
 from .wire_cutting_transforms import cut_wires, expand_observables
 from .automated_cut_finding import find_cuts, DeviceConstraints, OptimizationParameters
 
@@ -104,6 +107,7 @@
     "cut_gates",
     "generate_cutting_experiments",
     "reconstruct_expectation_values",
+    "reconstruct_distribution",
     "PartitionedCuttingProblem",
     "cut_wires",
     "expand_observables",

qiskit_addon_cutting/cutting_decomposition.py

@@ -189,10 +189,10 @@ def partition_problem(
     if observables is not None and any(obs.phase != 0 for obs in observables):
         raise ValueError("An input observable has a phase not equal to 1.")
 
-    if len(circuit.cregs) != 0 or circuit.num_clbits != 0:
-        raise ValueError(
-            "Circuits input to partition_problem should contain no classical registers or bits."
-        )
+    # if len(circuit.cregs) != 0 or circuit.num_clbits != 0:
+    #     raise ValueError(
+    #         "Circuits input to partition_problem should contain no classical registers or bits."
+    #     )
 
     # Determine partition labels from connectivity (ignoring TwoQubitQPDGates)
     # if partition_labels is not specified

qiskit_addon_cutting/cutting_experiments.py

@@ -35,7 +35,7 @@
 
 def generate_cutting_experiments(
     circuits: QuantumCircuit | dict[Hashable, QuantumCircuit],
-    observables: PauliList | dict[Hashable, PauliList],
+    observables: PauliList | dict[Hashable, PauliList] | None,
     num_samples: int | float,
 ) -> tuple[
     list[QuantumCircuit] | dict[Hashable, list[QuantumCircuit]],