remove unused lines from svae.py

svae/svae.py

@@ -3,11 +3,10 @@
 from util import add, sub, contract, scale, unbox
 
 
-def make_gradfun(run_inference, recognize, loglike, eta_prior, return_flat=False):
+def make_gradfun(run_inference, recognize, loglike, eta_prior):
     saved = lambda: None
 
     def mc_vlb(eta, phi, psi, y_n, N, L):
-        T = y_n.shape[0]
         nn_potentials = recognize(y_n, psi)
         samples, stats, global_vlb, local_vlb = run_inference(
             eta_prior, eta, nn_potentials, num_samples=L)
@@ -20,11 +19,4 @@ def gradfun(y_n, N, L, eta, phi, psi):
         eta_natgrad = sub(add(eta_prior, saved.stats), eta)
         return vlb, (eta_natgrad, phi_grad, psi_grad)
 
-    def flat_gradfun(y_n, N, L, eta, phi, psi):
-        objective = lambda (eta, phi, psi): mc_vlb(eta, phi, psi, y_n, N, L)
-        return vgrad(objective)((eta, phi, psi))
-
-    if return_flat:
-        return gradfun, flat_gradfun
-
     return gradfun