class_docstr: Tests for uncertainty prediction.
class_docstr_tok: ["Tests","for","uncertainty","prediction","."]
docstr: Test predict_r_uncertain.
docstr_tok: ["Test","predict_r_uncertain","."]
code: def test_predict_r_uncertain(self):
      """Test predict_r_uncertain."""
      self._test_predict_uncertain(
          method=predict_r_uncertain,
          expected_shape=(self.batch_size, self.factory.num_relations),
          ht_batch=self.batch[:, [0, 2]],
      )
  
file_path: pykeen/tests/test_uncertainty.py
imports: from typing import Callable, Tuple
  import torch
  from pykeen.models import ERMLPE, TransE
  from pykeen.models.uncertainty import UncertainPrediction, predict_h_uncertain, predict_hrt_uncertain, predict_r_uncertain, predict_t_uncertain
  from tests import cases
label_desc: Use image node_id 4 for calling the UncertaintyPredictionTestCase obj's underlying member method code with example usage: obj.test_predict_r_uncertain() without return types