Flyte enables production-grade orchestration for machine learning workflows and data processing created to accelerate local workflows to production.

Flyte

<https://github.com/flyteorg/flytekit-python-template/pull/31|#31 verification was failing in GPU mode [mnist-training]>
Pull request opened by <https://github.com/zeryx|zeryx>
Verification step failed when `gpu_enabled` was set to True, fixed by adding an additional conditional to dataset.
<https://github.com/flyteorg/flytekit-python-template|flyteorg/flytekit-python-template>
:white_check_mark: All checks have passed
2/2 successful checks

<https://github.com/flyteorg/flytekit-python-template/pull/31|#31 verification was failing in GPU mode [mnist-training]>
Pull request closed by <https://github.com/zeryx|zeryx>
<https://github.com/flyteorg/flytekit-python-template|flyteorg/flytekit-python-template>