Merge pull request #662 from jeverink/Printing-of-HybridGibbs

Print for HybridGibbs

Author: amal-ghamdi

cuqi/distribution/_joint_distribution.py

@@ -283,7 +283,7 @@ def __repr__(self):
             msg += f"\t{density.name} ~ {density}\n"
 
         # Wrap up
-        msg += ")"
+        msg += " )"
 
         return msg
 

cuqi/experimental/mcmc/_gibbs.py

@@ -359,3 +359,16 @@ def _store_samples(self):
         """ Store current samples at index i of samples dict """
         for par_name in self.par_names:
             self.samples[par_name].append(self.current_samples[par_name])
+
+    def __repr__(self):
+        """ Return a string representation of the sampler. """
+        msg = f"Sampler: {self.__class__.__name__} \n"
+        if self.target is None:
+            msg += f" Target: None \n"
+        else:
+            msg += f" Target: \n \t {self.target} \n\n"
+            
+        for key, value in zip(self.samplers.keys(), self.samplers.values()):
+            msg += f" Variable '{key}' with {value} \n"
+
+        return msg