code: def test_consume_scores(num_entities: int, num_relations: int):
      """Test for consume_scores."""
      dataset = pykeen.predict.AllPredictionDataset(
          num_entities=num_entities, num_relations=num_relations
      )
      model = pykeen.models.mocks.FixedModel(
          triples_factory=KGInfo(
              num_entities=num_entities,
              num_relations=num_relations,
              create_inverse_triples=False,
          )
      )
      consumer = pykeen.predict.CountScoreConsumer()
      pykeen.predict.consume_scores(model, dataset, consumer)
      assert consumer.batch_count == num_relations * num_entities
      assert consumer.score_count == num_relations * num_entities**2
  
imports: from typing import Any, Collection, Iterable, MutableMapping, Optional, Sequence, Tuple, Union
  import numpy
  import pandas
  import pytest
  import torch
  import unittest_templates
  import pykeen.models
  import pykeen.models.mocks
  import pykeen.predict
  import pykeen.regularizers
  import pykeen.typing
  from pykeen.constants import COLUMN_LABELS
  from pykeen.datasets.nations import Nations
  from pykeen.triples.triples_factory import AnyTriples, CoreTriplesFactory, KGInfo
  from pykeen.utils import resolve_device
  from tests import cases
label_desc: Use image node_id 2 for calling a global function with example usage: test_consume_scores(num_entities, num_relations) without return types