TESTDATA_PATH = flags.DEFINE_string(
    "path", None, help="Path to testdata directory.")

MODEL_NAME = flags.DEFINE_string(
    "model_name", None, help="Name of model to generate.")


class DataStructureModel(module.Module):
  """Model used for testing data structures in the C++ API."""

  def __init__(self):
    self.arr1 = [1.]
    self.const_arr = [constant_op.constant(1.)]