Update utils.py
Browse files
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 |
|