class_docstr: Test models are set in the right mode at the right time.
class_docstr_tok: ["Test","models","are","set","in","the","right","mode","at","the","right","time","."]
docstr: Check the scores produced by a forward function.
docstr_tok: ["Check","the","scores","produced","by","a","forward","function","."]
code: def _check_scores(self, scores) -> None:
      """Check the scores produced by a forward function."""
      # check for finite values by default
      assert torch.all(torch.isfinite(scores)).item()
  
imports: import unittest
  from dataclasses import dataclass
  from typing import Callable
  import torch
  from pykeen.datasets import Nations
  from pykeen.models import FixedModel, Model, TransE
  from pykeen.triples import TriplesFactory
  from pykeen.typing import COLUMN_HEAD, COLUMN_RELATION, COLUMN_TAIL, Target
  from pykeen.utils import extend_batch, resolve_device
label_desc: Use image node_id 2 for calling the TestBaseModel obj's underlying member method code with example usage: obj._check_scores(scores) without return types