code: def step(self, data_refs, validation_data_refs=None):
      from bigdl.orca.data.utils import partition_get_data_label
  
      partition_list = ray.get(data_refs)
      partition_data = [
          item for partition in partition_list for item in partition
      ]
      data, label = partition_get_data_label(
          partition_data, allow_tuple=True, allow_list=False
      )
      data_size = len(label)
      steps = data_size // self.config["batch_size"] + 1
      print("Number of train records for this worker: ", data_size)
      print("Number of steps for this worker: ", steps)
      train_dataset = to_tensor_slice_dataset(data, label, self.config)
  
      test_dataset = None
      if validation_data_refs:
          validation_partition_list = ray.get(validation_data_refs)
          validation_partition_data = [
              item
              for partition in validation_partition_list
              for item in partition
          ]
          validation_data, validation_label = partition_get_data_label(
              validation_partition_data,
              allow_tuple=True,
              allow_list=False,
          )
          test_data_size = len(validation_label)
          test_steps = test_data_size // self.config["batch_size"] + 1
          self.config["test_steps"] = test_steps
          print(
              "Number of test records for this worker: ", test_data_size
          )
          print(
              "Number of test steps for this worker: ",
              self.config["test_steps"],
          )
          test_dataset = to_tensor_slice_dataset(
              validation_data, validation_label, self.config
          )
  
      return main(
          train_dataset,
          test_dataset,
          self.tf_config,
          self.server,
          self.config,
      )
  
imports: import time
  import argparse
  import tensorflow
  import os
  import sys
  import math
  import collections
  from tensorflow.python.client import timeline
  import json
  from tensorflow.python.ops import partitioned_variables
  import ray
  from bigdl.dllib.utils.log4Error import *
  from bigdl.orca import init_orca_context, stop_orca_context, OrcaContext
  from bigdl.orca.data.utils import process_spark_xshards
  from bigdl.orca.learn.utils import maybe_dataframe_to_xshards
  from bigdl.friesian.feature import FeatureTable
label_desc: Use image node_id 5 for calling the RayWorker obj's underlying member method code with example usage: obj.step(data_refs, validation_data_refs) and returns: main