Add docstrings

Author: Bart van der Vecht

squidasm/run/stack/run.py

@@ -41,6 +41,7 @@
 class DepolariseWithFailureAnyBellStateSamplerFactory(
     HeraldedStateDeliverySamplerFactory
 ):
+    """State sampler that samples any of the 4 Bell states with equal probablity."""
     def __init__(self):
         super().__init__(func_delivery=self._delivery_func)
 
@@ -86,6 +87,8 @@ def _delivery_func(prob_max_mixed, prob_success, **kwargs):
 
 
 class DepolariseWithFailureAnyBellMagicDistributor(MagicDistributor):
+    """Distributor that creates any of the 4 Bell states with equal probablity."""
+
     def __init__(self, nodes, prob_max_mixed, prob_success, **kwargs):
         self.prob_max_mixed = prob_max_mixed
         self.prob_success = prob_success