oucgc1996 commited on
Commit
18ffc8d
1 Parent(s): b455269

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -1
utils.py CHANGED
@@ -200,7 +200,9 @@ def parsePDB(PDBFile,keep_only_chains=None,keep_hetatm=True,bb_only=False):
200
  atomNamesTMP += ["HET_"+str(resnum)+"_"+atnameHet+"_"+line[21]]
201
  coords+=[torch.tensor(coordsTMP)]
202
  atomNames += [atomNamesTMP]
203
- print(atomNames, pdbname, pdb_num)
 
 
204
  return torch.torch.nn.utils.rnn.pad_sequence(coords, batch_first=True, padding_value=PADDING_INDEX), atomNames, pdbname, pdb_num
205
 
206
 
 
200
  atomNamesTMP += ["HET_"+str(resnum)+"_"+atnameHet+"_"+line[21]]
201
  coords+=[torch.tensor(coordsTMP)]
202
  atomNames += [atomNamesTMP]
203
+ print(atomNames)
204
+ print(pdbname)
205
+ print(pdb_num)
206
  return torch.torch.nn.utils.rnn.pad_sequence(coords, batch_first=True, padding_value=PADDING_INDEX), atomNames, pdbname, pdb_num
207
 
208