From 9dc24c4f81cda0e5677da98e1f9f54b15cdc48c3 Mon Sep 17 00:00:00 2001 From: jyaacoub Date: Mon, 13 Nov 2023 12:52:25 -0500 Subject: [PATCH] fix(foldseek): dataset created successfully #53 --- src/feature_extraction/protein_nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/feature_extraction/protein_nodes.py b/src/feature_extraction/protein_nodes.py index 31c78d6d..eda4ae82 100644 --- a/src/feature_extraction/protein_nodes.py +++ b/src/feature_extraction/protein_nodes.py @@ -76,7 +76,7 @@ def get_foldseek_onehot(combined_seq): # the target sequence includes 3Di foldseek tokens alternating with the actual sequence # so we divide by 2 to get the length of the actual sequence fld_hot = np.zeros((len(combined_seq)//2, len(ResInfo.foldseek_tokens))) - for i in range(1, len(combined_seq), step=2): + for i in range(1, len(combined_seq), 2): fld_hot[i // 2,] = one_hot(combined_seq[i], ResInfo.foldseek_tokens) return fld_hot