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