| 'msa': | |
| Traceback (most recent call last): | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/data/infer_data_pipeline.py", line 211, in __getitem__ | |
| data, atom_array, _ = self.process_one( | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/data/infer_data_pipeline.py", line 114, in process_one | |
| InferenceMSAFeaturizer.make_msa_feature( | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/data/msa_featurizer.py", line 1144, in make_msa_feature | |
| msa_feats = InferenceMSAFeaturizer.get_inference_prot_msa_features_for_assembly( | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/data/msa_featurizer.py", line 1047, in get_inference_prot_msa_features_for_assembly | |
| msa_info = bioassembly[int(entity_id_list[0]) - 1][PROT_TYPE_NAME]["msa"] | |
| KeyError: 'msa' | |
| 'msa': | |
| Traceback (most recent call last): | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/data/infer_data_pipeline.py", line 211, in __getitem__ | |
| data, atom_array, _ = self.process_one( | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/data/infer_data_pipeline.py", line 114, in process_one | |
| InferenceMSAFeaturizer.make_msa_feature( | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/data/msa_featurizer.py", line 1144, in make_msa_feature | |
| msa_feats = InferenceMSAFeaturizer.get_inference_prot_msa_features_for_assembly( | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/data/msa_featurizer.py", line 1047, in get_inference_prot_msa_features_for_assembly | |
| msa_info = bioassembly[int(entity_id_list[0]) - 1][PROT_TYPE_NAME]["msa"] | |
| KeyError: 'msa' | |
| [Rank 0]7pzb_need_search_msa 'watermark': | |
| Traceback (most recent call last): | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/runner/inference.py", line 338, in infer_predict | |
| prediction, label_dict = runner.predict(data) | |
| File "/n/home09/zaixizhang/miniconda3/envs/pro_new/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context | |
| return func(*args, **kwargs) | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/runner/inference.py", line 158, in predict | |
| prediction, label_dict, _ = self.model( | |
| File "/n/home09/zaixizhang/miniconda3/envs/pro_new/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl | |
| return self._call_impl(*args, **kwargs) | |
| File "/n/home09/zaixizhang/miniconda3/envs/pro_new/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl | |
| return forward_call(*args, **kwargs) | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/model/protenix_edit.py", line 1041, in forward | |
| pred_dict, label_dict, log_dict, time_tracker = self.main_inference_loop( | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/model/protenix_edit.py", line 521, in main_inference_loop | |
| "watermark": _cat(pred_dicts, "watermark") | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/model/protenix_edit.py", line 508, in _cat | |
| return torch.cat([x[key] for x in dict_list], dim=0) | |
| File "/n/holylabs/LABS/mzitnik_lab/Users/zaixizhang/Protenix_new/protenix/model/protenix_edit.py", line 508, in <listcomp> | |
| return torch.cat([x[key] for x in dict_list], dim=0) | |
| KeyError: 'watermark' | |