990d954031a1121bc0b9ec1678c0b7db7cfaa374,tpot/operators/base.py,Operator,parameter_types,#Operator#,117

Before Change

            annotation = operator_parameters[param].annotation

            // Raise RuntimeError if a type is not annotated
            if annotation == Signature.empty:
                raise RuntimeError("Undocumented argument type for {} in operator {}".
                    format(param, self.sklearn_class.__class__.__name__))
            else:
                arg_types.append(annotation)

        return_type = Output_DF if self.root else pd.DataFrame

        return (arg_types, return_type)

After Change

            num_args = len(getargspec(self.preprocess_args).args[1:])  // Remove "self"

        // Make sure the class has been written properly
        if num_args != len(self.arg_types):
            raise RuntimeError(("{}"s arg_types does not correspond to the "
                                "arguments defined for itself".format(self.__name__)))

        // First argument is always a DataFrame
        arg_types = [pd.DataFrame] + list(self.arg_types)
        return_type = Output_DF if self.root else pd.DataFrame

        return (arg_types, return_type)

Instances