Skip to content
Snippets Groups Projects
Commit 01195fa1 authored by B13nch3n's avatar B13nch3n
Browse files

Only merge PAE fiels on validation

parent 83bfcc99
Branches
No related tags found
No related merge requests found
...@@ -263,15 +263,22 @@ def _get_associated_files(model_cif_file, assoc_dir): ...@@ -263,15 +263,22 @@ def _get_associated_files(model_cif_file, assoc_dir):
if dat_cat is None: if dat_cat is None:
continue continue
idxs = _get_indeces( idxs = _get_indeces(
dat_cat, ["archive_file_id", "file_format", "file_path"] dat_cat,
["archive_file_id", "file_content", "file_format", "file_path"],
) )
for row in dat_cat: for row in dat_cat:
if row[idxs["file_format"]] == "cif": if row[idxs["file_format"]] == "cif":
arc_id = row[idxs["archive_file_id"]] if row[idxs["file_content"]] == "local pairwise QA scores":
arc_file = archives[arc_id][0] arc_id = row[idxs["archive_file_id"]]
cif_file = row[idxs["file_path"]] arc_file = archives[arc_id][0]
data = _unzip_arc_cif(arc_file, cif_file, assoc_dir) cif_file = row[idxs["file_path"]]
assoc_files.append((data, archives[arc_id][1])) data = _unzip_arc_cif(arc_file, cif_file, assoc_dir)
assoc_files.append((data, archives[arc_id][1]))
elif row[idxs["file_content"]] not in ["other"]:
raise RuntimeError(
"Unknown associated CIF file content "
+ f"found: {row[idxs['file_content']]}"
)
return assoc_files, mdl_cif, entry_id_map return assoc_files, mdl_cif, entry_id_map
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment