andreyvelich: Fix Unit Tests

JobBuild
152b920b
09-15 23:43
b88482dc
09-15 23:22
93a58147
09-14 14:31
10c36fae
09-14 13:39
93d072b5
09-07 16:09
d3f22c82
09-07 14:28
kubeflow-pytorch-operator-presubmit 1306015641882333184 1306010358137229313 1305514367822860289 1305501280910708738 1304161825213911040 1303002329196269568 1302976913106538496