docs: update documentation

Author: nabil-Tounarti

pre-compute/src/compute/pre_compute_app.rs

@@ -38,15 +38,19 @@ impl PreComputeAppTrait for PreComputeApp {
     /// Runs the complete pre-compute pipeline.
     ///
     /// This method orchestrates the entire pre-compute process:
-    /// 1. Reads configuration arguments
-    /// 2. Validates the output folder exists
-    /// 3. Downloads and decrypts the dataset (if required)
-    /// 4. Downloads all input files
+    /// 1. Reads the output directory from environment variable `IEXEC_PRE_COMPUTE_OUT`
+    /// 2. Reads and validates configuration arguments from environment variables
+    /// 3. Validates the output folder exists
+    /// 4. Downloads and decrypts all datasets (if required)
+    /// 5. Downloads all input files
+    ///
+    /// The method collects all errors encountered during execution and returns them together,
+    /// allowing partial completion when possible (e.g., if one dataset fails, others are still processed).
     ///
     /// # Returns
     ///
     /// - `Ok(())` if all operations completed successfully
-    /// - `Err(ReplicateStatusCause)` if any step failed
+    /// - `Err(Vec<ReplicateStatusCause>)` containing all errors encountered during execution
     ///
     /// # Example
     ///
@@ -123,18 +127,19 @@ impl PreComputeAppTrait for PreComputeApp {
     /// Downloads the input files listed in `pre_compute_args.input_files` to the specified `output_dir`.
     ///
     /// Each URL is hashed (SHA-256) to generate a unique local filename.
-    /// If any download fails, the function returns an error.
+    /// The method continues downloading all files even if some downloads fail.
     ///
-    /// # Returns
+    /// # Behavior
     ///
-    /// - `Ok(())` if all files are downloaded successfully.
-    /// - `Err(ReplicateStatusCause::PreComputeInputFileDownloadFailed(url))` if any file fails to download.
+    /// - Downloads continue even when individual files fail
+    /// - Successfully downloaded files are saved with SHA-256 hashed filenames
+    /// - All download failures are collected and returned together
     ///
-    /// # Panics
+    /// # Returns
     ///
-    /// This function panics if:
-    /// - `pre_compute_args` is `None`.
-    /// - `chain_task_id` is `None`.
+    /// - `Ok(())` if all files are downloaded successfully
+    /// - `Err(Vec<ReplicateStatusCause>)` containing a `PreComputeInputFileDownloadFailed` error
+    ///   for each file that failed to download
     fn download_input_files(&self) -> Result<(), Vec<ReplicateStatusCause>> {
         let mut exit_causes: Vec<ReplicateStatusCause> = Vec::new();
         let args = &self.pre_compute_args;

pre-compute/src/compute/pre_compute_args.rs

@@ -23,10 +23,12 @@ pub struct PreComputeArgs {
 impl PreComputeArgs {
     /// Constructs a validated `PreComputeArgs` instance by reading and validating environment variables.
     ///
+    /// This method collects all errors encountered during validation instead of failing on the first error,
+    /// allowing for complete error reporting and partial processing where possible.
+    ///
     /// # Environment Variables
     /// This method reads the following environment variables:
     /// - Required for all tasks:
-    ///   - `IEXEC_PRE_COMPUTE_OUT`: Output directory path
     ///   - `IEXEC_DATASET_REQUIRED`: Boolean ("true"/"false") indicating dataset requirement
     ///   - `IEXEC_INPUT_FILES_NUMBER`: Number of input files to load
     ///   - `IEXEC_BULK_SLICE_SIZE`: Number of bulk datasets (0 means no bulk processing)
@@ -42,22 +44,22 @@ impl PreComputeArgs {
     ///   - `IEXEC_DATASET_#_KEY`: Dataset decryption key
     /// - Input file URLs (`IEXEC_INPUT_FILE_URL_1`, `IEXEC_INPUT_FILE_URL_2`, etc.)
     ///
-    /// # Errors
-    /// Returns `ReplicateStatusCause` error variants for:
-    /// - Missing required environment variables
-    /// - Invalid boolean values in `IEXEC_DATASET_REQUIRED`
-    /// - Invalid numeric format in `IEXEC_INPUT_FILES_NUMBER` or `IEXEC_BULK_SLICE_SIZE`
-    /// - Missing dataset parameters when required
-    /// - Missing input file URLs
-    /// - Missing bulk dataset parameters when bulk processing is enabled
+    /// # Returns
+    ///
+    /// Returns a tuple containing:
+    /// - `PreComputeArgs`: The constructed arguments (with `output_dir` set to empty string)
+    /// - `Vec<ReplicateStatusCause>`: A vector of all errors encountered (empty if successful)
     ///
     /// # Example
     ///
     /// ```rust
     /// use tee_worker_pre_compute::compute::pre_compute_args::PreComputeArgs;
     ///
-    /// // Typically called with task ID from execution context
-    /// let args = PreComputeArgs::read_args();
+    /// let (mut args, errors) = PreComputeArgs::read_args();
+    /// if !errors.is_empty() {
+    ///     eprintln!("Encountered {} error(s) while reading arguments", errors.len());
+    /// }
+    /// args.output_dir = "/path/to/output".to_string(); // Set output_dir separately
     /// ```
     pub fn read_args() -> (PreComputeArgs, Vec<ReplicateStatusCause>) {
         info!("Starting to read pre-compute arguments from environment variables");