code: def test_badge_with_classifier_that_does_not_return_embeddings_proba(
      self,
  ):
      # fake_embed return random embeddings and does not return probabilities
      def fake_embed(
          data_set, module_selector=lambda x: x["fc"], pbar="tqdm"
      ):
          return np.random.rand(len(data_set), 10)
  
      query_strategy = BADGE(6)
      with mock.patch.object(
          KimCNNClassifier, "embed", wraps=fake_embed
      ):
          self._simple_exhaustive_active_learning_test(query_strategy)
  
file_path: small-text/tests/integration/small_text/integrations/pytorch/query_strategies/test_query_strategies.py
imports: import unittest
  import pytest
  import numpy
  from unittest import mock
  from small_text.integrations.pytorch.exceptions import PytorchNotFoundError
  from tests.integration.small_text.query_strategies.test_query_strategies import QueryStrategiesExhaustiveIntegrationTest
label_desc: Use image node_id 8 for calling the QueryStrategiesTest obj's underlying member method code with example usage: obj.test_badge_with_classifier_that_does_not_return_embeddings_proba() and returns: np