NameError: name 'leaf_type' is not defined

[alisiahkoohi]

In line 119 of hint.py, leaf_type is passed as an optional argument to initialize the HierarchicalAffineCouplingTree class, as follows:

        self.tree = HierarchicalAffineCouplingTree(dims_in[0],
                                                   conv=conv,
                                                   subnet_constructor=subnet_constructor,
                                                   c_internal=c_internal,
                                                   clamp=clamp,
                                                   max_splits=max_splits,
                                                   min_split_size=min_split_size,
                                                   leaf_type=leaf_type)

HierarchicalAffineCouplingTree does not seem to be able to accept such an argument. Additionally, leaf_type that is being passed is not defined.

[psteinb]

I second this observation, commenting out leaf_type=leaf_type appears to make this error go away. Maybe this hints to experimental code, that was never committed nor pushed?