Fixed import of AUTOTUNE

tensorflow / similarity

TensorFlow Similarity is a python package focused on making similarity learning quick and easy.

Apache License 2.0

1.01k stars 104 forks source link

Fixed import of AUTOTUNE #288

Open arcAman07 opened 2 years ago

arcAman07 commented 2 years ago

I was running the pytest command at the root of the dir and I found an import error of AUTOTUNE. According to the TensorFlow 2.3 documentation, AUTOTUNE lies under the tensorflow.data.experimental rather tensorflow.data. This fixes that error. Signed off @arcAman07 Tagging the maintainers for review : @owenvallis

owenvallis commented 2 years ago

is tf.data.AUTOTUNE a breaking import in 2.3? The official docs import as tf.data.AUTOTUNE for current versions of TF. See here https://www.tensorflow.org/guide/data_performance#prefetching