Update docs remove print.

recirq/qcqmc/analysis.py

@@ -68,7 +68,10 @@ def _json_dict_(self):
 
 @attrs.frozen(eq=False)
 class OverlapAnalysisData(data.Data):
-    """Container for analysis of overlap.
+    """Container for storing the (shadow tomography-)reconstructed wavefunctions.
+
+    See the factory method `build_analysis_from_dependencies` which will
+    properly construct this object from the (real or simulated) experimental bitstrings.
 
     Args:
         params: The parameters for overlap construction.
@@ -92,6 +95,9 @@ def build_analysis_from_dependencies(
     ) -> "OverlapAnalysisData":
         """Builds a OverlapAnalysisData from OverlapAnalysisParams.
 
+        Given the sampled bitstrings from either a real experiment or a
+        simulated experiment, this will construct the shadow wavefunction.
+
         Args:
             params: The parameters for overlap analysis.
             dependencies: The dependencies leading up to this point (i.e. the
@@ -134,7 +140,7 @@ def get_variational_energy(
         hamiltonian_data: hamiltonian.HamiltonianData,
         k: Union[str, int] = 1,
     ) -> float:
-        """Gets the variational energy of the reconstructed wavefunction.
+        """Computes the variational energy of the reconstructed wavefunction.
 
         Args:
             analysis_data: The overlap analysis data.

recirq/qcqmc/analysis_test.py

@@ -127,15 +127,6 @@ def test_small_experiment_partition_match_reversed_partition_same_seed(
         k=1,
     )
 
-    print(energy_1)
-    print(energy_2)
-    print(trial_wf.ansatz_energy)
-    print(trial_wf.ansatz_energy - energy_1)
-    print(trial_wf.ansatz_energy - energy_2)
-
-    print(analysis_1.reconstructed_wf_for_k["1"])
-    print(analysis_2.reconstructed_wf_for_k["1"])
-
     assert np.abs(energy_1 - energy_2) < target_error