code: def test_that_logits_get_converted_to_probabilities(logits):
      logits = logits.numpy()
  
      # define a random func:
      @requireprobs
      def wrapped(_, logits):
          return logits
  
      probability_distribution = wrapped(None, logits)
      assert np.alltrue(
          (probability_distribution >= 0)
          & (probability_distribution <= 1)
      ).all()
  
imports: import warnings
  import numpy
  import pytest
  from hypothesis import given
  from torch_hypothesis import class_logits
  from baal.active import get_heuristic
  from baal.active.heuristics import Random, BALD, Margin, Entropy, Certainty, Variance, BatchBALD, AbstractHeuristic, requireprobs, Precomputed, CombineHeuristics
  from tests.test_utils import make_fake_dist, make_3d_fake_dist, make_5d_fake_dist
label_desc: Use image node_id 11 for calling a global function with example usage: test_that_logits_get_converted_to_probabilities(logits) and returns: logits