corrected shift function

dd0d7bab · Kaan Güney Keklikçi · 323654ae · dd0d7bab
Commit dd0d7bab authored Aug 09, 2021 by Kaan Güney Keklikçi
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

maf.py scripts/flows/maf/maf.py +1 -1

No files found.
--- a/scripts/flows/maf/maf.py
+++ b/scripts/flows/maf/maf.py
@@ -68,7 +68,7 @@ class MAF(object):
    def make_maf(self, data):
        distribution = self.base_dist
        sample_shape = self.get_dims(data)
-        shift_scale_function = self.get_shift_scale_func(data)
+        shift_scale_function = self.get_shift_scale_func()
        bijector = tfb.MaskedAutoregressiveFlow(shift_scale_function)
        maf = tfd.TransformedDistribution(tfd.Sample(distribution, sample_shape), bijector)
        return maf