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 |
|