Update code/add_structure.py

code/add_structure.py

@@ -156,11 +156,11 @@ def get_pdb_ids(protein_id):
     try:
         request = IdMappingClient.submit(
             source="UniProtKB_AC-ID", dest="PDB", ids={protein_id})
-        pdb_list = list(request.each_result())
-        return [i['to'] for i in pdb_list]
-    except IdMappingError:
-        print('HERE')
-        get_pdb_ids_2(protein_id)
+        try:
+            pdb_list = list(request.each_result())
+            return [i['to'] for i in pdb_list]
+        except:
+            get_pdb_ids_2(protein_id)
     except requests.exceptions.HTTPError:
         get_pdb_ids_2(protein_id)
     except KeyError: