From 09e51ca37f0e21d234e14c188738d01a19f549de Mon Sep 17 00:00:00 2001 From: mkohle02 Date: Fri, 17 Apr 2026 19:07:08 -0400 Subject: [PATCH] change np.isnan to pd.isna --- modelgenerator/structure_tokenizer/datasets/protein_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelgenerator/structure_tokenizer/datasets/protein_dataset.py b/modelgenerator/structure_tokenizer/datasets/protein_dataset.py index 125d0c4..b280acf 100644 --- a/modelgenerator/structure_tokenizer/datasets/protein_dataset.py +++ b/modelgenerator/structure_tokenizer/datasets/protein_dataset.py @@ -163,7 +163,7 @@ def from_config( def __getitem__(self, index: int) -> dict[str, torch.Tensor | str | None]: row = self.proteins_df.iloc[index] filename = row["filename"] - chain = row["chain"] if not np.isnan(row["chain"]) else "nan" + chain = row["chain"] if not pd.isna(row["chain"]) else "nan" match Path(filename).suffix: case ".cif.gz" | ".cif": entity = row["entity"]