add depends_on step

Signed-off-by: tylertitsworth <tyler.titsworth@intel.com>

Author: tylertitsworth

jax/docker-compose.yaml

@@ -65,6 +65,8 @@ services:
       target: jax-base
     command: >
       bash -c "python -c 'import jax; print(\"Jax Version:\", jax.__version__)'"
+    depends_on:
+      - ${PACKAGE_OPTION:-pip}
     image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-${BASE_IMAGE_NAME:-ubuntu}-${BASE_IMAGE_TAG:-22.04}-${PACKAGE_OPTION:-pip}-py${PYTHON_VERSION:-3.10}-xla-${INTEL_XLA_VERSION:-v0.4.0}-base
     pull_policy: always
   jupyter: