code: def train_split():
      # pylint: disable=unused-argument
      def func(dataset, y):
          ds_train = type(dataset)(dataset.X[:2], dataset.y[:2])
          ds_valid = type(dataset)(dataset.X[2:], dataset.y[2:])
          return ds_train, ds_valid
  
      return func
  
file_path: skorch/skorch/tests/conftest.py
imports: import numpy
  import pytest
  from sklearn.datasets import make_classification
  from sklearn.datasets import make_regression
  from sklearn.preprocessing import StandardScaler
  import torch
  from torch import nn
label_desc: Use image node_id 9 for calling a global function with example usage: train_split() and returns: func, ds_train,  ds_valid