added e2e test to workflow

Signed-off-by: Shashank Mittal <[email protected]>
kubeflow · Sep 2, 2024 · af7ec73 · af7ec73
1 parent e1ab55b
commit af7ec73
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/.github/workflows/e2e-test-pytorch-mnist.yaml b/.github/workflows/e2e-test-pytorch-mnist.yaml
@@ -40,6 +40,7 @@ jobs:
  # suggestion-hyperopt
  - "long-running-resume,from-volume-resume,median-stop"
  # others
+ - "grid,bayesian-optimization,tpe,multivariate-tpe,cma-es,hyperband"
  - "hyperopt-distribution"
  - "file-metrics-collector,pytorchjob-mnist"
- - "median-stop-with-json-format,file-metrics-collector-with-json-format"
+ - "median-stop-with-json-format,file-metrics-collector-with-json-format"
diff --git a/examples/v1beta1/hp-tuning/hyperopt-distribution.yaml b/examples/v1beta1/hp-tuning/hyperopt-distribution.yaml
@@ -20,11 +20,14 @@ spec:
  feasibleSpace:
  min: "0.01"
  max: "0.05"
+ step: "0.01"
+ distribution: "uniform"
  - name: momentum
  parameterType: double
  feasibleSpace:
  min: "0.5"
  max: "0.9"
+ distribution: "logUniform"
  trialTemplate:
  primaryContainerName: training-container
  trialParameters:
@@ -54,4 +57,4 @@ spec:
  limits:
  memory: "1Gi"
  cpu: "0.5"
- restartPolicy: Never
+ restartPolicy: Never