class_docstr: Tests for MarginalDistributionBaseline.
code: def _verify(self, scores: torch.FloatTensor):  # noqa: D102
      # check probability distribution
      assert (0.0 <= scores).all() and (scores <= 1.0).all()
      assert torch.allclose(
          scores.sum(dim=1), torch.ones(self.batch_size)
      )
  
file_path: pykeen/tests/test_baselines.py
imports: import torch
  import pykeen.models
  from tests import cases
label_desc: Use image node_id 1 for calling the MarginalDistributionBaselineTests obj's underlying member method code with example usage: obj._verify(scores) without return types