Skip to content

Commit c791831

Browse files
nutsiepullytensorflower-gardener
authored andcommitted
Move quantize_functional_test target to PY3
PiperOrigin-RevId: 317247999
1 parent cb52f5d commit c791831

File tree

1 file changed

+1
-2
lines changed
  • tensorflow_model_optimization/python/core/quantization/keras

1 file changed

+1
-2
lines changed

tensorflow_model_optimization/python/core/quantization/keras/BUILD

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -302,12 +302,11 @@ py_test(
302302
],
303303
)
304304

305-
# TODO(pulkitb): Using PY2 since MNIST data downloading code throws an SSL error on Sponge.
306305
py_test(
307306
name = "quantize_functional_test",
308307
size = "large",
309308
srcs = ["quantize_functional_test.py"],
310-
python_version = "PY2",
309+
python_version = "PY3",
311310
# To match parallel runs of run_all_keras_modes.
312311
shard_count = 4,
313312
deps = [

0 commit comments

Comments
 (0)