Fix vestigial calls to apply.__default_rule__ (#336)

Author: eb8680

effectful/handlers/numpyro.py

@@ -89,7 +89,7 @@ def _apply(op, *args, **kwargs):
         typ = op.__type_rule__(*args, **kwargs)
         if issubclass(typ, dist.Distribution):
             return defdata(op, *args, **kwargs)
-        return apply.__default_rule__({}, op, *args, **kwargs)
+        return op.__default_rule__(*args, **kwargs)
 
     with runner({apply: _apply}):
         d = defterm(d)
@@ -137,7 +137,7 @@ def _apply(op, *args, **kwargs):
         typ = op.__type_rule__(*args, **kwargs)
         if issubclass(typ, dist.Distribution):
             return defdata(op, *args, **kwargs)
-        return apply.__default_rule__({}, op, *args, **kwargs)
+        return op.__default_rule__(*args, **kwargs)
 
     with runner({apply: _apply}):
         d = defterm(d)

effectful/handlers/pyro.py

@@ -377,7 +377,7 @@ def _apply(op, *args, **kwargs):
             typ, pyro.distributions.torch_distribution.TorchDistributionMixin
         ):
             return defdata(op, *args, **kwargs)
-        return apply.__default_rule__({}, op, *args, **kwargs)
+        return op.__default_rule__(*args, **kwargs)
 
     with runner({apply: _apply}):
         d = defterm(d)
@@ -425,7 +425,7 @@ def _apply(op, *args, **kwargs):
             typ, pyro.distributions.torch_distribution.TorchDistributionMixin
         ):
             return defdata(op, *args, **kwargs)
-        return apply.__default_rule__({}, op, *args, **kwargs)
+        return op.__default_rule__(*args, **kwargs)
 
     with runner({apply: _apply}):
         d = defterm(d)