code: def get_proto_mol(atoms):
      """ """
      mol = Chem.MolFromSmarts("[#" + str(atoms[0]) + "]")
      rwMol = Chem.RWMol(mol)
      for i in range(1, len(atoms)):
          a = Chem.Atom(int(atoms[i]))
          rwMol.AddAtom(a)
  
      mol = rwMol.GetMol()
  
      return mol
  
imports: from rdkit import Chem
  import numpy
  import networkx
  import collections
  from collections import defaultdict
  import itertools
  import copy
  from scipy.spatial import distance_matrix
label_desc: Use image node_id 2 for calling a global function with example usage: get_proto_mol(atoms) and returns: mol