Match sklearn signatures