diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
index ea9e1923..fadae569 100644
--- a/.github/workflows/docs.yml
+++ b/.github/workflows/docs.yml
@@ -33,7 +33,7 @@ jobs:
     - name: pip install
       run: |
         python -m pip install --upgrade pip==21.3.1
-        python -m pip install markdown-include==0.6.0 mkdocstrings==0.15.0 mkdocs-material==7.1.3 livereload==2.6.3
+        python -m pip install markdown-include==0.6.0 mkdocstrings==0.17.0 mkdocs-material==7.1.3 livereload==2.6.3
         
         # NOTE: we need requirements to be able to parse reference docs scripts
         sudo apt-get install libopenmpi-dev
diff --git a/requirements.txt b/requirements.txt
index d2534489..8ac0b66c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,6 +16,8 @@ flaml==0.9.6
 # pipelines
 shrike[pipeline]==1.14.7
 azure-ml-component==0.9.4.post1  # for component dsl
+jinja2==2.11.3 #for docs
+MarkupSafe==1.1.1
 azureml-train-core==1.36.0  # for azureml.train.hyperdrive
 azureml-dataset-runtime==1.36.0  # to register dataset
 hydra-core~=1.0.3