fixed test

RECETOX · Aug 16, 2024 · ae94362 · ae94362
1 parent b07150b
commit ae94362
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/tools/ms2deepscore/macros.xml b/tools/ms2deepscore/macros.xml
@@ -55,7 +55,7 @@
 
     <xml name="config_generator">
         <section name="model_structure" title="Model Structure" expanded="true">
-            <repeat name="layers" title="Layer" min="1" default="3" >
+            <repeat name="layers" title="Layer" min="1" default="1" >
                 <param name="dims" type="integer" label="Dimensions" min = "0" value="2000" help="Size of the in-between layer to add." />
             </repeat>
             <param name="embedding_dim" type="integer" label="Embedding Dimension" value="400" help="The dimension of the final embedding layer." />

diff --git a/tools/ms2deepscore/ms2deepscore_config_generator.xml b/tools/ms2deepscore/ms2deepscore_config_generator.xml
@@ -60,9 +60,8 @@ settings.save_to_file("$output_file")
 
     <tests>
         <test expect_num_outputs="1">
-            <param name="layers_0|dims" value="2000"/>
-            <param name="layers_1|dims" value="2000"/>
-            <param name="layers_2|dims" value="2000"/>
+            <param name="layers_0|dims" value="20"/>
+            <param name="layers_1|dims" value="20"/>
             <param name="embedding_dim" value="15" />
             <param name="ionisation_mode" value="negative" />
             <param name="epochs" value="2" />