From 935f9aa62591bb52c0bf359214b53a881e3e9aed Mon Sep 17 00:00:00 2001 From: v-shaoningli Date: Thu, 29 Jun 2023 00:02:35 -0700 Subject: [PATCH] add except in sabdab --- diffab/datasets/sabdab.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/diffab/datasets/sabdab.py b/diffab/datasets/sabdab.py index 4b96c1b..e9238fc 100644 --- a/diffab/datasets/sabdab.py +++ b/diffab/datasets/sabdab.py @@ -139,8 +139,13 @@ def preprocess_sabdab_structure(task): pdb_path = task['pdb_path'] parser = PDB.PDBParser(QUIET=True) - model = parser.get_structure(id, pdb_path)[0] - + + try: + model = parser.get_structure(id, pdb_path)[0] + except Exception as e: + logging.warning(f'Failed to parse {pdb_path}: {e}') + return None + parsed = { 'id': entry['id'], 'heavy': None,