code: def main():
      mnist_dir = get_mnist_dir()
      run(data_dir=mnist_dir)
  
file_path: petastorm/examples/spark_dataset_converter/pytorch_converter_example.py
imports: from __future__ import division
  import logging
  import torch
  import torch.nn
  import torch.nn.functional
  import torch.optim
  from pyspark.sql import SparkSession
  from torch.autograd import Variable
  from examples.spark_dataset_converter.utils import get_mnist_dir
  from petastorm.spark import SparkDatasetConverter, make_spark_converter
label_desc: Use image node_id 4 for calling a global function with example usage: main() without return types