Hey I'm trying to run the pytorch distributed mnis...
# ask-the-community
c
Hey I'm trying to run the pytorch distributed mnist example locally: https://docs.flyte.org/en/latest/flytesnacks/examples/kfpytorch_plugin/pytorch_mnist.html The training seems to go fine but getting the following error:
Copy code
Failed with Unknown Exception <class 'TypeError'> Reason: Error encountered while executing 'pytorch_training_wf':
  Failed to convert outputs of task 'main.mnist_pytorch_job' at position model_state:
  Incorrect file type, expected application/octet-stream, got application/zip
Seems like it's due to this line: https://github.com/flyteorg/flytekit/blob/69a3218abbc787936074b17bd09435cc48e62596/flytekit/types/file/file.py#L348 Replacing
PythonPickledFile
with
FlyteFile
skips the filetype check and just works
n
@Cornelis Boon thanks for finding this bug! would you mind updating the docs with this fix? https://github.com/flyteorg/flytesnacks/blob/master/examples/kfpytorch_plugin/kfpytorch_plugin/pytorch_mnist.py#L195
c