oucgc1996 commited on
Commit
df23a61
1 Parent(s): 1afa9ba

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +0 -3
utils.py CHANGED
@@ -179,9 +179,6 @@ def parsePDB(PDBFile,keep_only_chains=None,keep_hetatm=True,bb_only=False):
179
  atomNamesTMP += ["HET_"+str(resnum)+"_"+atnameHet+"_"+line[21]]
180
  coords+=[torch.tensor(coordsTMP)]
181
  atomNames += [atomNamesTMP]
182
- print(atomNames)
183
- print(pdbname)
184
- print(pdb_num)
185
  return torch.torch.nn.utils.rnn.pad_sequence(coords, batch_first=True, padding_value=PADDING_INDEX), atomNames, pdbname, pdb_num
186
 
187
 
 
179
  atomNamesTMP += ["HET_"+str(resnum)+"_"+atnameHet+"_"+line[21]]
180
  coords+=[torch.tensor(coordsTMP)]
181
  atomNames += [atomNamesTMP]
 
 
 
182
  return torch.torch.nn.utils.rnn.pad_sequence(coords, batch_first=True, padding_value=PADDING_INDEX), atomNames, pdbname, pdb_num
183
 
184