From 66e3e4eaad399afd5843495a0aff197e6c778ef0 Mon Sep 17 00:00:00 2001
From: Seth Michael Larson <seth.larson@elastic.co>
Date: Tue, 2 Nov 2021 09:09:25 -0500
Subject: [PATCH] Set 'script.max_compilations_rate: use-context'

---
 docs/requirements-docs.txt | 9 ++++++++-
 tests/setup_tests.py       | 7 ++++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/docs/requirements-docs.txt b/docs/requirements-docs.txt
index 3f9b956..8e47875 100644
--- a/docs/requirements-docs.txt
+++ b/docs/requirements-docs.txt
@@ -1,4 +1,11 @@
--r ../requirements-dev.txt
+elasticsearch>=7.7
+pandas>=1.2.0
+matplotlib
+nbval
+numpydoc>=0.9.0
+scikit-learn>=0.22.1
+xgboost>=1
+lightgbm
 nbsphinx
 numpydoc>=0.9.0
 git+https://github.com/pandas-dev/pydata-sphinx-theme.git@master
diff --git a/tests/setup_tests.py b/tests/setup_tests.py
index dcea39c..228e423 100644
--- a/tests/setup_tests.py
+++ b/tests/setup_tests.py
@@ -90,7 +90,12 @@ def _update_max_compilations_limit(es, limit="10000/1m"):
     if es_version(es) < (7, 8):
         body = {"transient": {"script.max_compilations_rate": limit}}
     else:
-        body = {"transient": {"script.context.field.max_compilations_rate": limit}}
+        body = {
+            "transient": {
+                "script.max_compilations_rate": "use-context",
+                "script.context.field.max_compilations_rate": limit,
+            }
+        }
     es.cluster.put_settings(body=body)