Ensure index arrays have type int32

  l: list

    List of integers. Typically small counts.

  """

  return np.delete(np.insert(np.cumsum(l), 0, 0), -1) + offset

  return np.delete(np.insert(np.cumsum(l, dtype=np.int32), 0, 0), -1) + offset

def cumulative_sum(l, offset=0):

    ]

    # Convert to numpy array

    self.deg_block_indices = np.array(deg_block_indices)

    self.deg_block_indices = np.array(deg_block_indices, dtype=np.int32)

  def get_atoms_with_deg(self, deg):

    """Retrieves atom_features with the specific degree"""

      to_cat = [self.canon_adj_list[i] for i in indices]

      if len(to_cat) > 0:

        adj_list = np.vstack([self.canon_adj_list[i] for i in indices])

        self.deg_adj_lists[deg - self.min_deg] = adj_list

        self.deg_adj_lists[deg - self.min_deg] = adj_list.astype(np.int32)

      else:

        self.deg_adj_lists[deg - self.min_deg] = np.zeros(

        for deg in range(min_deg, max_deg + 1)

    ]

    # Update the old adjcency lists with the new atom indices and then combine

    # Update the old adjacency lists with the new atom indices and then combine

    # all together

    for deg in range(min_deg, max_deg + 1):

      row = 0  # Initialize counter