fixed model tests

fratajcz · Oct 14, 2024 · ea27750 · ea27750
1 parent 80bc9ec
commit ea27750
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/speos/preprocessing/handler.py b/speos/preprocessing/handler.py
@@ -11,7 +11,7 @@ def __init__(self, config, preprocessor_kwargs):
 
         """
         self.config = config
-        mappings = GWASMapper().get_mappings(config.input.tag, fields=config.input.field)
+        mappings = GWASMapper(self.config.input.gwas_mappings).get_mappings(config.input.tag, fields=config.input.field)
 
         adjacencies = AdjacencyMapper(config.input.adjacency_mappings, blacklist=self.config.input.adjacency_blacklist).get_mappings(config.input.adjacency, fields=config.input.adjacency_field)
 

diff --git a/speos/tests/test_models.py b/speos/tests/test_models.py
@@ -20,6 +20,7 @@ def setUp(self):
         super().setUp()
 
         self.config.name = "SimpleModelTest"
+        self.config.input.tag = "dummy"
         self.config.training.pos_weight = 2
         self.config.training.dilution = 2
         #self.config.model.mp.type = "gcn"