sinter 1.12.dev1688842788__tar.gz → 1.12.dev1688897006__tar.gz

{sinter-1.12.dev1688842788 → sinter-1.12.dev1688897006}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sinter
-Version: 1.12.dev1688842788
+Version: 1.12.dev1688897006
 Summary: Samples stim circuits and decodes them using pymatching.
 Author: Craig Gidney
 Author-email: craig.gidney@gmail.com

{sinter-1.12.dev1688842788 → sinter-1.12.dev1688897006}/setup.py

@@ -19,7 +19,7 @@ with open('README.md', encoding='UTF-8') as f:
 with open('requirements.txt', encoding='UTF-8') as f:
     requirements = [line.split()[0] for line in f.read().splitlines()]
 
-__version__ = '1.12.dev1688842788'
+__version__ = '1.12.dev1688897006'
 
 setup(
     name='sinter',

{sinter-1.12.dev1688842788 → sinter-1.12.dev1688897006}/src/sinter/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '1.12.dev1688842788'
+__version__ = '1.12.dev1688897006'
 
 from sinter._anon_task_stats import (
     AnonTaskStats,

{sinter-1.12.dev1688842788 → sinter-1.12.dev1688897006}/src/sinter/_plotting.py

@@ -426,7 +426,7 @@ def plot_custom(
         kwargs.setdefault('color', 'black')
         ax.plot(xs, ys, **kwargs)
 
-        if line_fits is not None and len(xs) >= 2:
+        if line_fits is not None and len(set(xs)) >= 2:
             x_scale, y_scale = line_fits
             fit_xs = _rescale(xs, x_scale, False)
             fit_ys = _rescale(ys, y_scale, False)

{sinter-1.12.dev1688842788 → sinter-1.12.dev1688897006}/src/sinter/_probability_util.py

@@ -1,7 +1,7 @@
 import dataclasses
 import math
 import pathlib
-from typing import Any, Dict, Union, Callable, Sequence, TYPE_CHECKING
+from typing import Any, Dict, Union, Callable, Sequence, TYPE_CHECKING, overload
 
 import numpy as np
 
@@ -363,11 +363,18 @@ def fit_binomial(
     return Fit(best=num_hits / num_shots, low=low / num_shots, high=high / num_shots)
 
 
+@overload
 def shot_error_rate_to_piece_error_rate(shot_error_rate: float, *, pieces: float, values: float = 1) -> float:
+    pass
+@overload
+def shot_error_rate_to_piece_error_rate(shot_error_rate: 'sinter.Fit', *, pieces: float, values: float = 1) -> 'sinter.Fit':
+    pass
+def shot_error_rate_to_piece_error_rate(shot_error_rate: Union[float, 'sinter.Fit'], *, pieces: float, values: float = 1) -> Union[float, 'sinter.Fit']:
     """Convert from total error rate to per-piece error rate.
 
     Args:
-        shot_error_rate: The rate at which shots fail.
+        shot_error_rate: The rate at which shots fail. If this is set to a sinter.Fit,
+            the conversion broadcasts over the low,best,high of the fit.
         pieces: The number of xor-pieces we want to subdivide each shot into,
             as if each piece was an independent chance for the shot to fail and
             the total chance of a shot failing was the xor of each piece
@@ -418,6 +425,14 @@ def shot_error_rate_to_piece_error_rate(shot_error_rate: float, *, pieces: float
         ... )
         0.12052311142021144
     """
+
+    if isinstance(shot_error_rate, Fit):
+        return Fit(
+            low=shot_error_rate_to_piece_error_rate(shot_error_rate=shot_error_rate.low, pieces=pieces, values=values),
+            best=shot_error_rate_to_piece_error_rate(shot_error_rate=shot_error_rate.best, pieces=pieces, values=values),
+            high=shot_error_rate_to_piece_error_rate(shot_error_rate=shot_error_rate.high, pieces=pieces, values=values),
+        )
+
     if not (0 <= shot_error_rate <= 1):
         raise ValueError(f'need (0 <= shot_error_rate={shot_error_rate} <= 1)')
     if pieces <= 0:

{sinter-1.12.dev1688842788 → sinter-1.12.dev1688897006}/src/sinter.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sinter
-Version: 1.12.dev1688842788
+Version: 1.12.dev1688897006
 Summary: Samples stim circuits and decodes them using pymatching.
 Author: Craig Gidney
 Author-email: craig.gidney@gmail.com