code: def test_discriminator_pass(
      enhancer: SpectrogramEnhancerModel, sample_input
  ):
      input_spectrograms, lengths = sample_input
      input_spectrograms = rearrange(
          input_spectrograms, "b c l -> b 1 c l"
      )
      logits = enhancer.discriminator(
          x=input_spectrograms,
          condition=input_spectrograms,
          lengths=lengths,
      )
  
      assert logits.shape == lengths.shape
  
file_path: NeMo/tests/collections/tts/test_spectrogram_enhancer.py
imports: import pytest
  import torch
  from einops import rearrange
  from omegaconf import DictConfig
  from nemo.collections.tts.models import SpectrogramEnhancerModel
  from nemo.collections.tts.parts.utils.helpers import mask_sequence_tensor
label_desc: Use image node_id 11 for calling a global function with example usage: test_discriminator_pass(enhancer, sample_input) without return types