code: def _get_steps(data, window_size=4):
      """Extract the steps dataset and create out of it a window dataset."""
      episode_ds = data["steps"]
      # The line below creates a variant dataset
      return episode_ds.window(window_size, drop_remainder=True)
  
imports: import functools
  import pickle
  import numpy
  import pytest
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 17 for calling a global function with example usage: _get_steps(data, window_size) and returns: episode_ds