fix sparse sampling step in runs

BIONF · Sep 9, 2024 · a91f9e3 · a91f9e3
1 parent 2282b7f
commit a91f9e3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "XspecT"
-version = "0.2.4"
+version = "0.2.5"
 description = "Tool to monitor and characterize pathogens using Bloom filters."
 readme = {file = "README.md", content-type = "text/markdown"}
 license = {file = "LICENSE"}

diff --git a/src/xspect/models/probabilistic_filter_svm_model.py b/src/xspect/models/probabilistic_filter_svm_model.py
@@ -107,7 +107,7 @@ def predict(
         ),
         filter_ids: list[str] = None,
         step: int = 1,
-    ) -> dict:
+    ) -> ModelResult:
         """Predict the labels of the sequences"""
         # get scores and format them for the SVM
         res = super().predict(sequence_input, filter_ids, step=step)
@@ -119,6 +119,7 @@ def predict(
             self.slug(),
             res.hits,
             res.num_kmers,
+            sparse_sampling_step=step,
             prediction=str(svm.predict(svm_scores)[0]),
         )