Warning, /AID2E-framework/examples/basic/input-joblib.yml is written in an unsupported language. File is not indexed.
0001 # Minimal DTLZ2 + Ax + JobLib input for
0002 # `aid2e optimize examples/basic/input-joblib.yml`.
0003 #
0004 # This is intentionally small so it can be used as the first CLI smoke test.
0005 # Uses a top-level workflow so it can run through the generic CLI runner.
0006
0007 problem:
0008 name: "DTLZ2 CLI JobLib Smoke Test"
0009 problem_type: "DTLZ2"
0010 output_location: "/tmp"
0011 work_location: "/tmp"
0012 design_parameters_file: "../optimizers/dtlz2_design.params"
0013 objectives:
0014 - name: "f1"
0015 direction: "minimize"
0016 objective_plan:
0017 steps:
0018 stages:
0019 - name: "evaluate_dtlz2"
0020 script:
0021 path: "examples/complete/scripts/slurm_dtlz2_from_cli.py"
0022 output_file: "objectives.json"
0023 timeout_sec: 300
0024 produces_objective: true
0025 metrics_keys: ["f1"]
0026 - name: "f2"
0027 direction: "minimize"
0028 objective_plan:
0029 steps:
0030 stages:
0031 - name: "evaluate_dtlz2"
0032 script:
0033 path: "examples/complete/scripts/slurm_dtlz2_from_cli.py"
0034 output_file: "objectives.json"
0035 timeout_sec: 300
0036 produces_objective: true
0037 metrics_keys: ["f2"]
0038
0039 optimizer:
0040 name: "ax"
0041 type: "bayesian"
0042 parameters:
0043 initialization_strategy: "sobol"
0044 generator: "BOTORCH_MODULAR"
0045 n_initial_samples: 1
0046 n_iterations: 20
0047 batch_size: 1
0048 seed: 7
0049
0050 scheduler:
0051 runner_type: "JobLibRunner"
0052 parameters:
0053 n_jobs: 1
0054 backend: "threading"
0055 timeout: 300
0056 verbose: 0
0057 max_retries: 1
0058 output_location: "/tmp/aid2e_scheduler_output"
0059 monitor_interval: 5
0060
0061 workflows:
0062 workflows:
0063 - name: "dtlz2_joblib_eval"
0064 description: "Evaluate DTLZ2 objectives through a generic JobLib-backed command workflow."
0065 branches:
0066 - name: "main"
0067 stages:
0068 - name: "evaluate"
0069 parallelism:
0070 max_concurrent: 1
0071 retry_max: 0
0072 timeout_sec: 300
0073 jobs:
0074 - name: "dtlz2"
0075 command: >-
0076 {{python_executable}} {{repo_root}}/examples/complete/scripts/slurm_dtlz2_from_cli.py
0077 --x1 {{design_point[DTLZ2_variables.x1]}}
0078 --x2 {{design_point[DTLZ2_variables.x2]}}
0079 --x3 {{design_point[DTLZ2_variables.x3]}}
0080 --x4 {{design_point[DTLZ2_variables.x4]}}
0081 --x5 {{design_point[DTLZ2_variables.x5]}}
0082 --output {{output_dir}}/objectives.json
0083 --repeat 1
0084 payload:
0085 evaluator_type: "bash"
0086 outputs:
0087 - path: "{{output_dir}}/objectives.json"
0088 format: "json"
