From c0d371d8af5fd29e18c067ca9914e1163cc94c18 Mon Sep 17 00:00:00 2001 From: ntnn19 Date: Thu, 25 Jul 2024 13:45:21 +0200 Subject: [PATCH] Fixed bug in PR #633 --- colabfold/batch.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/colabfold/batch.py b/colabfold/batch.py index 00f393ed..602adcaf 100644 --- a/colabfold/batch.py +++ b/colabfold/batch.py @@ -1,3 +1,4 @@ + from __future__ import annotations import os @@ -373,6 +374,9 @@ def predict_structure( # TODO: add pad_input_mulitmer() input_features = feature_dict input_features["asym_id"] = input_features["asym_id"] - input_features["asym_id"][...,0] + input_features['chain_num'] = input_features["asym_id"].max() + 1 + model_runner.chain_num = input_features['chain_num'] + else: if model_num == 0: input_features = model_runner.process_features(feature_dict, random_seed=seed)