code: def test_task_type_same_with_model_or_y_pred(
      diabetes_split_dataset_and_model,
  ):
      # Arrange
      train, _, model = diabetes_split_dataset_and_model
      # Act
      ctx1 = Context(train, model=model)
      ctx2 = Context(
          train, y_pred_train=model.predict(train.features_columns)
      )
      # Assert
      assert ctx1.task_type == ctx2.task_type
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_task_type_same_with_model_or_y_pred(diabetes_split_dataset_and_model) without return types