From ea4df5c3c59943e96d04cbabe09b603b3a4aaa6e Mon Sep 17 00:00:00 2001
From: Robrecht Cannoodt <rcannood@gmail.com>
Date: Tue, 23 Apr 2024 07:12:41 +0200
Subject: [PATCH] extend api files

---
 src/task/api/comp_method_run.yaml   | 32 +++++++++++++++++++++++++++++
 src/task/api/comp_method_train.yaml | 28 +++++++++++++++++++++++++
 2 files changed, 60 insertions(+)
 create mode 100644 src/task/api/comp_method_run.yaml
 create mode 100644 src/task/api/comp_method_train.yaml

diff --git a/src/task/api/comp_method_run.yaml b/src/task/api/comp_method_run.yaml
new file mode 100644
index 00000000..ed38fb4c
--- /dev/null
+++ b/src/task/api/comp_method_run.yaml
@@ -0,0 +1,32 @@
+functionality:
+  namespace: "methods"
+  info:
+    type: method
+    type_info:
+      label: Run
+      summary: Apply a pre-trained model
+      description: |
+        Apply a pre-trained model for the DGE perturbation prediction task.
+  arguments:
+    - name: --de_train
+      __merge__: file_de_train_parquet.yaml
+      required: true
+      direction: input
+    - name: --id_map
+      __merge__: file_id_map.yaml
+      required: true
+      direction: input
+    - name: --model
+      type: file
+      description: The model to use for prediction.
+      required: true
+      direction: input
+    - name: --output
+      __merge__: file_prediction.yaml
+      required: true
+      direction: output
+  test_resources:
+    - type: python_script
+      path: /src/common/component_tests/run_and_check_output.py
+    - path: /resources/neurips-2023-data
+      dest: resources/neurips-2023-data
\ No newline at end of file
diff --git a/src/task/api/comp_method_train.yaml b/src/task/api/comp_method_train.yaml
new file mode 100644
index 00000000..14e5025e
--- /dev/null
+++ b/src/task/api/comp_method_train.yaml
@@ -0,0 +1,28 @@
+functionality:
+  namespace: "methods"
+  info:
+    type: method
+    type_info:
+      label: Train
+      summary: Train a model for a regression method.
+      description: |
+        Train a model for a regression method for the DGE perturbation prediction task.
+  arguments:
+    - name: --de_train
+      __merge__: file_de_train_parquet.yaml
+      required: true
+      direction: input
+    - name: --id_map
+      __merge__: file_id_map.yaml
+      required: true
+      direction: input
+    - name: --model
+      type: file
+      description: The model to use for prediction.
+      required: true
+      direction: output
+  test_resources:
+    - type: python_script
+      path: /src/common/component_tests/run_and_check_output.py
+    - path: /resources/neurips-2023-data
+      dest: resources/neurips-2023-data
\ No newline at end of file