message: Add a method for choosing a random TF dtype while ignoring a list of invalid ones.
diff: @@ -126,12 +126,19 @@ class FuzzingHelper(object):
  else:
  return self.get_float_list(min_length, max_length)
  - def get_tf_dtype(self):
  - """Return a random tensorflow type.
  + def get_tf_dtype(self, allowed_set=None):
  + """Return a random tensorflow dtype.
  +
  + Args:
  + allowed_set: An allowlisted set of dtypes to choose from instead of all of
  + them.
  Returns:
  A random type from the list containing all TensorFlow types.
  """
  + if allowed_set:
  + index = self.get_int(0, len(allowed_set) - 1)
  + else:
  index = self.get_int(0, len(_TF_DTYPES) - 1)
  return _TF_DTYPES[index]
  