Fix distort_image_ops_test

[WindQAQ]

Fix failed builds because of update in upstream.

[kyleabeauchamp]

Is there a nightly build link or something that shows the failure? I trust you, but for the sake of reproducibility might be nice to have some sort of link to the exception.

[kyleabeauchamp]

LGTM

[WindQAQ]

@kyleabeauchamp, hi Kyle, here it is.
https://source.cloud.google.com/results/invocations/0388ed59-b166-4bdd-85e3-cb05ac762e22/targets

[kyleabeauchamp]

Thanks!

[WindQAQ]

Shape checking within tf.function is somewhat not robust in either c++ or python API... @facaiy, hi facai, is there any suggested way to do this?

Btw, thanks for asking and review, Kyle 😄

[seanpmorgan]

Thanks for the fix! Just trying to understand what happend here... typically the RegEx is supposed to catch the error thrown by this I believe:
https://github.com/tensorflow/addons/blob/master/tensorflow_addons/custom_ops/image/cc/ops/distort_image_ops.cc#L33

But at first pass I can't find the commit that affected this. LGTM to merge if @facaiy is good with it.. but I think something has changed with shape checking in our op?

[seanpmorgan]

One option would be to merge this fix and create a separate issue to look into why this happend.

[WindQAQ]

Thanks for the fix! Just trying to understand what happend here... typically the RegEx is supposed to catch the error thrown by this I believe:
https://github.com/tensorflow/addons/blob/master/tensorflow_addons/custom_ops/image/cc/ops/distort_image_ops.cc#L33

But at first pass I can't find the commit that affected this. LGTM to merge if @facaiy is good with it.. but I think something has changed with shape checking in our op?

Thank you, Sean! The original regex text and the type of exception seem wrong while the test passed previously...

addons/tensorflow_addons/custom_ops/image/cc/kernels/adjust_hsv_in_yiq_op.cc

Lines 57 to 59 in d46dba1

	OP_REQUIRES(context, input.dims() >= 3,
	errors::InvalidArgument("input must be at least 3-D, got shape",
	input.shape().DebugString()));

Anyway, because the tests fail randomly, I will create an issue for it.

[seanpmorgan]

Approving.. discussion of how this should be done can be solved in #260