code: def create_discriminator(data_input_shape, conditional_input_shape):
      data_input_shape = data_input_shape[0]
      conditional_input_shape = conditional_input_shape[0]
  
      return nn.Sequential(
          Discriminator(data_input_shape, conditional_input_shape)
      )
  
file_path: deepchem/deepchem/models/torch_models/tests/test_gan.py
imports: import deepchem
  import numpy
  import pytest
  import tempfile
  from flaky import flaky
label_desc: Use image node_id 2 for calling a global function with example usage: create_discriminator(data_input_shape, conditional_input_shape) and returns: nn