code: def test_partially_restricted_prediction_dataset(
      heads, relations, tails, target
  ):
      """Test for PartiallyRestrictedPredictionDataset."""
      ds = pykeen.predict.PartiallyRestrictedPredictionDataset(
          heads=heads, relations=relations, tails=tails, target=target
      )
      # try accessing each element
      for i in range(len(ds)):
          _ = ds[i]
  
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 15 for calling a global function with example usage: test_partially_restricted_prediction_dataset(heads, relations, tails, target) without return types