Fix results logging and linting

- Updated logging logic to correctly append verification results to `results.log` instead of overwriting.
- Cleaned up linting issues (line length, trailing whitespace).

Co-authored-by: refraction-ray <35157286+refraction-ray@users.noreply.github.com>

Author: google-labs-jules[bot]

examples/reproduce_papers/2020_efficient_classical_simulation_random_shallow_2d/main.py

@@ -356,6 +356,13 @@ def standardize_tensor(site_data):
 
 
 def main():
+    # Clear log file
+    with open(
+        "examples/reproduce_papers/2020_efficient_classical_simulation_random_shallow_2d/outputs/results.log",
+        "w",
+    ) as f:
+        f.write("Verification Results:\n")
+
     # 1. Verification on Small Grids
     logger.info("Running verification on small grids...")
 
@@ -381,9 +388,14 @@ def main():
             end = time.time()
 
             diff = abs(exact_prob - sebd_prob)
-            logger.info(
-                f"Bitstring: {target_bits[:5]}... Exact: {exact_prob:.6e}, SEBD: {sebd_prob:.6e}, Diff: {diff:.2e}"
-            )
+            msg = f"Bitstring: {target_bits[:5]}... Exact: {exact_prob:.6e}, SEBD: {sebd_prob:.6e}, Diff: {diff:.2e}"
+            logger.info(msg)
+
+            with open(
+                "examples/reproduce_papers/2020_efficient_classical_simulation_random_shallow_2d/outputs/results.log",
+                "a",
+            ) as f:
+                f.write(f"Grid {rows}x{cols} depth {depth}: {msg}\n")
 
             # Relaxed threshold for larger grids/floating point error
             if diff > 1e-8:
@@ -409,7 +421,7 @@ def main():
     # Save results
     with open(
         "examples/reproduce_papers/2020_efficient_classical_simulation_random_shallow_2d/outputs/results.log",
-        "w",
+        "a",
     ) as f:
         f.write(f"Large Scale 10x10:\nProb: {prob}\nTime: {end - start:.4f}s\n")
 

examples/reproduce_papers/2020_efficient_classical_simulation_random_shallow_2d/outputs/results.log

@@ -1,3 +1,10 @@
+Verification Results:
+Grid 2x2 depth 2: Bitstring: 1101... Exact: 3.177146e-03, SEBD: 3.177145e-03, Diff: 9.72e-10
+Grid 2x2 depth 2: Bitstring: 0101... Exact: 3.501271e-03, SEBD: 3.501269e-03, Diff: 2.11e-09
+Grid 4x4 depth 4: Bitstring: 01000... Exact: 7.651079e-06, SEBD: 7.651004e-06, Diff: 7.45e-11
+Grid 4x4 depth 4: Bitstring: 11001... Exact: 1.286861e-05, SEBD: 1.286850e-05, Diff: 1.10e-10
+Grid 4x5 depth 4: Bitstring: 11010... Exact: 1.020886e-06, SEBD: 1.020875e-06, Diff: 1.02e-11
+Grid 4x5 depth 4: Bitstring: 00010... Exact: 5.336154e-07, SEBD: 5.336101e-07, Diff: 5.29e-12
 Large Scale 10x10:
 Prob: 4.330939486047805e-36
-Time: 66.6539s
+Time: 66.3942s