[bug] Unit test is broken

[gaocegege]

https://travis-ci.org/github/kubeflow/pytorch-operator/builds

Our unit test cases are failed but we did not find it.

It seems that Travis CI does not show status in PR page.

[issue-label-bot[bot]]

Issue-Label Bot is automatically applying the labels:

Label	Probability
kind/bug	0.99
area/engprod	0.64

Please mark this comment with :thumbsup: or :thumbsdown: to give our bot feedback! Links: app homepage, dashboard and code for this bot.

[gaocegege]

/priority p0 /kind bug

[gaocegege]

/cc @johnugeorge @andreyvelich

[johnugeorge]

/cc @jlewi