complex featurizer output fix

    if not isinstance(complexes, Iterable):

      complexes = [cast(Tuple[str, str], complexes)]

    features, failures = [], []

    features, failures, successes = [], [], []

    for idx, point in enumerate(complexes):

      if idx % log_every_n == 0:

        logger.info("Featurizing datapoint %i" % idx)

      try:

        features.append(self._featurize(point))

        successes.append(idx)

      except:

        logger.warning(

            "Failed to featurize datapoint %i. Appending empty array." % idx)

        failures.append(idx)

    # Find a successful featurization

    try:

      i = np.argmax([f.shape[0] for f in features])

      dtype = features[i].dtype

      shape = features[i].shape

      dummy_array = np.zeros(shape, dtype=dtype)

    except AttributeError:

      dummy_array = features[successes[0]]

    # Replace failed featurizations with appropriate array

    for idx in failures: