examples/pytorch_mnist/testing/pytorchjob_test.py

16 lines
295 B
Python

"""Test training using PytorchJob.
This file tests that we can submit the job from ksonnet
and that the job runs to completion.
It is an integration test as it depends on having access to
a Kubeflow deployment to submit the PytorchJob to.
TODO Implement tests
"""
def test_train():
pass