code: def get_train_val_data(train_size=1000, valid_size=400):
      data = get_x_y(size=train_size)
      validation_data = get_x_y(size=valid_size)
      return data, validation_data
  
imports: import os
  from unittest import TestCase
  import numpy
  import pytest
  import torch
  import torch.nn
  from torch.utils.data import Dataset, DataLoader
  from bigdl.orca.automl.auto_estimator import AutoEstimator
  from bigdl.orca.automl.pytorch_utils import LR_NAME
  from bigdl.orca.automl import hp
label_desc: Use image node_id 6 for calling a global function with example usage: get_train_val_data(train_size, valid_size) and returns: data,  validation_data