code: def test_batch_bald_fails_on_5d(distributions, reduction):
      np.random.seed(1338)
  
      bald = BatchBALD(100, reduction=reduction)
      with pytest.raises(ValueError) as e:
          marg = bald(distributions)
      assert "BatchBALD only works on classification" in str(e.value)
  
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 4 for calling a global function with example usage: test_batch_bald_fails_on_5d(distributions, reduction) without return types