class_docstr: Base class for contexts.
class_docstr_tok: ["Base","class","for","contexts","."]
docstr: Return the relevant Dataset by given kind.
docstr_tok: ["Return","the","relevant","Dataset","by","given","kind","."]
code: def get_data_by_kind(self, kind: DatasetKind):
      """Return the relevant Dataset by given kind."""
      if kind == DatasetKind.TRAIN:
          return self.train
      elif kind == DatasetKind.TEST:
          return self.test
      else:
          raise DeepchecksValueError(f"Unexpected dataset kind {kind}")
  
file_path: deepchecks/deepchecks/core/context.py
imports: from abc import ABC, abstractmethod
  from deepchecks.core import CheckFailure, CheckResult, DatasetKind
  from deepchecks.core.errors import DeepchecksNotSupportedError, DeepchecksValueError, ModelValidationError
label_desc: Use image node_id 6 for calling the BaseContext obj's underlying member method code with example usage: obj.get_data_by_kind(kind) and returns: self, self