code: def call_knn(self, func, inputs, batch_size):
      embeddings = self.get_embeddings_from_tensor_or_dataset(
          inputs, batch_size
      )
      func(embeddings)
  
imports: import numpy
  import torch
  from ..distances import BatchedDistance, CosineSimilarity
  from .None import common_functions
label_desc: Use image node_id 5 for calling the InferenceModel obj's underlying member method code with example usage: obj.call_knn(func, inputs, batch_size) without return types