code: def make_get_step_udf(multi_steps: Optional[int]):
      """Get step count by taking length of next_states_features array."""
  
      def get_step(col: List):
          return (
              1 if multi_steps is None else min(len(col), multi_steps)
          )
  
      return udf(get_step, LongType())
  
imports: import logging
  from typing import List, Optional, Tuple
  from pyspark.sql.functions import col, crc32, explode, map_keys, udf
  from pyspark.sql.types import ArrayType, BooleanType, FloatType, LongType, MapType, StructField, StructType
  from reagent.data.data_fetcher import DataFetcher
  from reagent.data.spark_utils import get_spark_session, get_table_url
  from reagent.workflow.types import Dataset, TableSpec
label_desc: Use image node_id 6 for calling a global function with example usage: make_get_step_udf(multi_steps) and returns: udf, unknown