add unit tests to train methods

openproblems-bio · Aug 26, 2024 · da30b86 · da30b86
1 parent 852c928
commit da30b86
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/api/comp_method_train.yaml b/src/api/comp_method_train.yaml
@@ -22,4 +22,11 @@ arguments:
   - name: "--output"
     __merge__: file_pretrained_model.yaml
     direction: output
-    required: true
+    required: true
+test_resources:
+  - type: python_script
+    path: /common/component_tests/check_config.py
+  - type: python_script
+    path: /common/component_tests/run_and_check_output.py
+  - path: /resources_test/predict_modality/openproblems_neurips2021/bmmc_cite/swap
+    dest: resources_test/predict_modality/openproblems_neurips2021/bmmc_cite/swap