Temporarily continue passing explain.mode for backwards compatibility

aws · Sep 18, 2024 · 3c8b6f9 · 3c8b6f9
1 parent 3d2bfd6
commit 3c8b6f9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/graph_notebook/neptune/client.py b/src/graph_notebook/neptune/client.py
@@ -444,7 +444,10 @@ def _gremlin_query_plan(self, query: str, plan_type: str, args: dict, ) -> reque
             data['language'] = 'gremlin'
             headers['content-type'] = 'application/json'
             if plan_type == 'explain':
-                data['explain-mode'] = args.pop('explain.mode')
+                # Remove explain.mode once HTTP is changed
+                explain_mode = args.pop('explain.mode')
+                data['explain.mode'] = explain_mode
+                data['explain-mode'] = explain_mode
             elif plan_type == 'profile':
                 for param, value in args.items():
                     data[param] = value