From 192f9d3991e175d1fa332b24ce17b8f33ba2bf49 Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Tue, 31 Mar 2020 11:04:56 +0200
Subject: [PATCH] fix typo

---
 pts/modules/flows.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pts/modules/flows.py b/pts/modules/flows.py
index 8ef12dd..b00ab58 100644
--- a/pts/modules/flows.py
+++ b/pts/modules/flows.py
@@ -275,7 +275,7 @@ class MADE(nn.Module):
 
     @property
     def base_dist(self):
-        return D.Normal(self.base_dist_mean, self.base_dist_var)
+        return Normal(self.base_dist_mean, self.base_dist_var)
 
     def forward(self, x, y=None):
         # MAF eq 4 -- return mean and log std