From 4abda1ae509063b1029ca92597a3a5e191d55e76 Mon Sep 17 00:00:00 2001
From: Dustin Tran <dustinviettran@gmail.com>
Date: Sun, 7 Jan 2018 22:33:38 -0800
Subject: [PATCH] fix pep8 from #808

---
 edward/inferences/hmc.py       | 14 +++++++-------
 edward/inferences/inference.py | 17 ++++++++++-------
 2 files changed, 17 insertions(+), 14 deletions(-)

diff --git a/edward/inferences/hmc.py b/edward/inferences/hmc.py
index 710dae95f..b96c43987 100644
--- a/edward/inferences/hmc.py
+++ b/edward/inferences/hmc.py
@@ -128,7 +128,7 @@ def build_update(self):
     # Update Empirical random variables.
     assign_ops = []
     for z_unconstrained, qz_unconstrained in six.iteritems(
-        self.latent_vars_unconstrained):
+            self.latent_vars_unconstrained):
       variable = qz_unconstrained.get_variables()[0]
       assign_ops.append(tf.scatter_update(
           variable, self.t, sample[z_unconstrained]))
@@ -139,7 +139,7 @@ def build_update(self):
 
   def _log_joint_unconstrained(self, z_sample):
     """
-    Given a sample in unconstrained latent space, transform it back into 
+    Given a sample in unconstrained latent space, transform it back into
     the original space, and compute the log joint density with appropriate
     Jacobian correction.
     """
@@ -151,17 +151,17 @@ def _log_joint_unconstrained(self, z_sample):
     z_sample_transformed = {}
     log_det_jacobian = 0.0
     for z_unconstrained, qz_unconstrained in z_sample.items():
-      z = (unconstrained_to_z[z_unconstrained] 
-           if z_unconstrained in unconstrained_to_z 
+      z = (unconstrained_to_z[z_unconstrained]
+           if z_unconstrained in unconstrained_to_z
            else z_unconstrained)
 
       try:
         bij = self.transformations[z].bijector
         z_sample_transformed[z] = bij.inverse(qz_unconstrained)
         log_det_jacobian += tf.reduce_sum(
-          bij.inverse_log_det_jacobian(qz_unconstrained))
-      except: # if z not in self.transformations, 
-              # or is not a TransformedDist w/ bijector
+            bij.inverse_log_det_jacobian(qz_unconstrained))
+      except:  # if z not in self.transformations,
+               # or is not a TransformedDist w/ bijector
         z_sample_transformed[z] = qz_unconstrained
 
     return self._log_joint(z_sample_transformed) + log_det_jacobian
diff --git a/edward/inferences/inference.py b/edward/inferences/inference.py
index 85dc7f116..a7cea84d2 100644
--- a/edward/inferences/inference.py
+++ b/edward/inferences/inference.py
@@ -15,6 +15,7 @@
 
 from tensorflow.contrib.distributions import bijectors
 
+
 @six.add_metaclass(abc.ABCMeta)
 class Inference(object):
   """Abstract base class for inference. All inference algorithms in
@@ -222,11 +223,13 @@ def initialize(self, n_iter=1000, n_print=None, scale=None,
     self.transformations = {}
     if auto_transform:
       latent_vars = self.latent_vars.copy()
-      self.latent_vars = {} # maps original latent vars to constrained Q's
-      self.latent_vars_unconstrained = {} # maps unconstrained vars to unconstrained Q's
+      # latent_vars maps original latent vars to constrained Q's.
+      # latent_vars_unconstrained maps unconstrained vars to unconstrained Q's.
+      self.latent_vars = {}
+      self.latent_vars_unconstrained = {}
       for z, qz in six.iteritems(latent_vars):
         if hasattr(z, 'support') and hasattr(qz, 'support') and \
-              z.support != qz.support and qz.support != 'point':
+                z.support != qz.support and qz.support != 'point':
 
           # transform z to an unconstrained space
           z_unconstrained = transform(z)
@@ -243,12 +246,12 @@ def initialize(self, n_iter=1000, n_print=None, scale=None,
           # back into the original constrained space
           if z_unconstrained != z:
             qz_constrained = transform(
-              qz_unconstrained, bijectors.Invert(z_unconstrained.bijector))
+                qz_unconstrained, bijectors.Invert(z_unconstrained.bijector))
 
-            try: # attempt to pushforward the params of Empirical distributions
+            try:  # attempt to pushforward the params of Empirical distributions
               qz_constrained.params = z_unconstrained.bijector.inverse(
-                qz_unconstrained.params)
-            except: # qz_unconstrained is not an Empirical distribution
+                  qz_unconstrained.params)
+            except:  # qz_unconstrained is not an Empirical distribution
               pass
 
           else: