1c10f0dc589fc13e011d8ffc61f3d94c8d6adbb8,pymanopt/core/problem.py,Problem,_flatten_arguments,#Problem#,99

Before Change

            if isinstance(group, (list, tuple)):
                flattened_arguments.extend(arguments[i])
            else:
                flattened_arguments.append(arguments[i])
        return flattened_arguments

    def _group_return_values(self, function, signature):
        Wraps a function inside another function which groups the return

After Change

        flattened_arguments = []
        for i, group_size in enumerate(signature):
            argument = arguments[i]
            if group_size == 1:
                assert not isinstance(argument, (list, tuple))
                flattened_arguments.append(argument)
            else:
                assert len(argument) == group_size
                flattened_arguments.extend(argument)
        return flattened_arguments

    def _group_return_values(self, function, signature):
        Wraps a function inside another function which groups the return

Instances