Update code/add_3Dalignment.py

code/add_3Dalignment.py

@@ -228,8 +228,8 @@ def get_alignments_3D(identifier, model_num, pdb_path, pdbSequence, source, chai
 
             atoms = [i.split() for i in atoms]
             atoms = [i for i in atoms if (i[2] == 'CA' and i[4]  == chain)]  
-            print('I am the problem its me')
-            print(atoms)
+            atoms = [[x[i][-3:] if i == 3 else x[i] for i in range(len(x))] for x in atoms]
+           
             atomSequence = ''.join([threeToOne(i[3]) for i in atoms])
             coords = [[i[6] ,i[7] ,i[8]] for i in atoms]
             resnums_for_sasa = [i[5] for i in atoms]