Clean ups (#315)

* Clean ups * pylint fix * formatting fix * reformat Co-authored-by: Rasul Karimov <[email protected]>
pymc-devs · Aug 28, 2020 · 42642ef · 42642ef
1 parent 0f16a47
commit 42642ef
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 372 deletions.
diff --git a/pymc4/distributions/continuous.py b/pymc4/distributions/continuous.py
@@ -13,7 +13,6 @@
     BoundedContinuousDistribution,
 )
 from pymc4.distributions import transforms
-from .half_student_t import HalfStudentT as TFPHalfStudentT
 
 
 __all__ = [
@@ -297,7 +296,7 @@ def __init__(self, name, df, scale, **kwargs):
     def _init_distribution(conditions, **kwargs):
         scale = conditions["scale"]
         df = conditions["df"]
-        return TFPHalfStudentT(df=df, loc=0, scale=scale, **kwargs)
+        return tfd.HalfStudentT(df=df, loc=0, scale=scale, **kwargs)
 
 
 class Beta(UnitContinuousDistribution):

diff --git a/pymc4/distributions/half_student_t.py b/pymc4/distributions/half_student_t.py
diff --git a/pymc4/flow/executor.py b/pymc4/flow/executor.py
@@ -5,7 +5,6 @@
 
 import tensorflow as tf
 
-import pymc4 as pm
 from pymc4 import coroutine_model
 from pymc4 import scopes
 from pymc4 import utils
@@ -351,7 +350,7 @@ def validate_return_object(self, return_object: Any):
             )
 
     def validate_return_value(self, return_value: Any):
-        pm.utils.map_nested(self.validate_return_object, return_value)
+        tf.nest.map_structure(self.validate_return_object, return_value)
 
     def evaluate_model(
         self,