--- a/wine_quality_train_eval.xml	Wed Dec 04 15:24:43 2024 +0000
+++ b/wine_quality_train_eval.xml	Wed Dec 04 18:34:56 2024 +0000
@@ -19,6 +19,8 @@
   <command detect_errors="aggressive">
     <![CDATA[
     . /fidle-tp/fidle-env/bin/activate;
+    mkdir /fidle-tp/fidle-env/cache_dir; export TRANSFORMERS_CACHE=/fidle-tp/fidle-env/cache_dir;
+    mkdir /fidle-tp/fidle-env/mpl_dir; export MPLCONFIGDIR=/fidle-tp/fidle-env/mpl_dir;
     /bin/bash -c "python3 /fidlemore/model_wine_lightning/wine_quality_train_eval.py -dataset_filepath ${dataset_csv}";
     cp OUTPUTS/model.ckpt ${model_ckpt};
     cp OUTPUTS/norm_config.json ${norm_config_json};

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/wine_quality_train_eval.xml.essai	Wed Dec 04 18:34:56 2024 +0000
@@ -0,0 +1,91 @@
+<tool id="wine_quality_train_eval" name="wine_quality_train_eval"
+ version="1.0.0">
+
+  <description>wine_quality_train_eval</description>
+
+  <requirements>
+
+    <!-- singularity -->
+    <container type="singularity">oras://registry.forgemia.inra.fr/nathalie.rousse/use/dnn_fidlemore:6d159f79349e231deadb738f79bd293c1a8dadd3</container>
+
+    <!-- image locale
+    <container type="singularity">/home/nrousse/workspace_git/SIWAA_regroup/USE_branch_dnn/use/DNN/containers/fidlemore.simg</container -->
+
+  </requirements>
+
+  <environment_variables>
+    <environment_variable name="FIDLE_DATASETS_DIR">/fidle-tp/datasets-fidle</environment_variable>
+  </environment_variables>
+  <command detect_errors="aggressive">
+    <![CDATA[
+    . /fidle-tp/fidle-env/bin/activate;
+    /bin/bash -c "mkdir ./cache_dir; export TRANSFORMERS_CACHE=./cache_dir";
+    /bin/bash -c "mkdir ./mpl_dir; export MPLCONFIGDIR=./mpl_dir";
+    /bin/bash -c "python3 /fidlemore/model_wine_lightning/wine_quality_train_eval.py -dataset_filepath ${dataset_csv}";
+    cp OUTPUTS/model.ckpt ${model_ckpt};
+    cp OUTPUTS/norm_config.json ${norm_config_json};
+    cp OUTPUTS/report.json ${report_json};
+    ]]>
+  </command>
+
+  <inputs>
+    <param name="dataset_csv" optional="true" type="data" format="csv"
+           label="File of dataset used to train and test the model (.csv)"/>
+  </inputs>
+
+  <outputs>
+    <data format="ckpt" name="model_ckpt"
+          label="model_ckpt model file (.ckpt)"/>
+    <data format="json" name="norm_config_json"
+          label="norm_config normalization configuration (.json)"/>
+    <data format="json" name="report_json" label="report (.json)"/>
+  </outputs>
+
+  <tests>
+  </tests>
+
+  <help><![CDATA[
+
+wine_quality_train_eval.xml
+
+===========================
+
+Code:
+
+----
+
+- wine_quality_train_eval.py
+
+Inputs:
+
+-------
+
+- dataset file (.csv) :
+
+  File containing dataset used to train and test the model.
+
+  The dataset will be splitted in 2 parts (training part, validation part).
+  
+  If not given, an embedded default file used.
+
+Outputs:
+
+--------
+
+- Model file (.ckpt)
+
+- Normalization configuration file (.json)
+
+- Report file (.json)
+
+Credits:
+
+--------
+
+-  Author: Nathalie Rousse nathalie.rousse@inrae.fr
+-  Copyright: INRAE
+
+  ]]>
+  </help>
+
+</tool>