code: def split_data(self):
      X = self.raw_df.drop("Wine", axis=1)
      y = self.raw_df[["Wine"]]
      (
          self.X_train,
          self.X_test,
          self.y_train,
          self.y_test,
      ) = train_test_split(
          X, y, test_size=self.test_size, random_state=self.seed
      )
      self.next(self.train)
  
file_path: wandb/tests/functional_tests/t0_main/metaflow/wandb-example-flow-decostep.py
imports: import os
  import pandas
  from metaflow import FlowSpec, Parameter, step
  from sklearn.ensemble import RandomForestClassifier
  from sklearn.metrics import accuracy_score
  from sklearn.model_selection import train_test_split
  from wandb.integration.metaflow import wandb_log
label_desc: Use image node_id 2 for calling the WandbExampleFlowDecoStep obj's underlying member method code with example usage: obj.split_data() without return types