pyro/infer/enum.py

Survived

Mutant 227

--- pyro/infer/enum.py
+++ pyro/infer/enum.py
@@ -56,7 +56,7 @@
     guide_trace.compute_score_parts()
     if is_validation_enabled():
         for site in model_trace.nodes.values():
-            if site["type"] == "sample":
+            if site["type"] == "XXsampleXX":
                 check_site_shape(site, max_plate_nesting)
         for site in guide_trace.nodes.values():
             if site["type"] == "sample":

Mutant 230

--- pyro/infer/enum.py
+++ pyro/infer/enum.py
@@ -59,7 +59,7 @@
             if site["type"] == "sample":
                 check_site_shape(site, max_plate_nesting)
         for site in guide_trace.nodes.values():
-            if site["type"] == "sample":
+            if site["type"] == "XXsampleXX":
                 check_site_shape(site, max_plate_nesting)
 
     return model_trace, guide_trace

Mutant 231

--- pyro/infer/enum.py
+++ pyro/infer/enum.py
@@ -108,7 +108,7 @@
     return partial(_config_fn, default, expand, num_samples, tmc)
 
 
-def config_enumerate(guide=None, default="parallel", expand=False, num_samples=None, tmc="diagonal"):
+def config_enumerate(guide=None, default="XXparallelXX", expand=False, num_samples=None, tmc="diagonal"):
     """
     Configures enumeration for all relevant sites in a guide. This is mainly
     used in conjunction with :class:`~pyro.infer.traceenum_elbo.TraceEnum_ELBO`.

Mutant 232

--- pyro/infer/enum.py
+++ pyro/infer/enum.py
@@ -108,7 +108,7 @@
     return partial(_config_fn, default, expand, num_samples, tmc)
 
 
-def config_enumerate(guide=None, default="parallel", expand=False, num_samples=None, tmc="diagonal"):
+def config_enumerate(guide=None, default="parallel", expand=True, num_samples=None, tmc="diagonal"):
     """
     Configures enumeration for all relevant sites in a guide. This is mainly
     used in conjunction with :class:`~pyro.infer.traceenum_elbo.TraceEnum_ELBO`.

Mutant 233

--- pyro/infer/enum.py
+++ pyro/infer/enum.py
@@ -108,7 +108,7 @@
     return partial(_config_fn, default, expand, num_samples, tmc)
 
 
-def config_enumerate(guide=None, default="parallel", expand=False, num_samples=None, tmc="diagonal"):
+def config_enumerate(guide=None, default="parallel", expand=False, num_samples=None, tmc="XXdiagonalXX"):
     """
     Configures enumeration for all relevant sites in a guide. This is mainly
     used in conjunction with :class:`~pyro.infer.traceenum_elbo.TraceEnum_ELBO`.