From 5a999ca85cf1163db0e60e9b46aad7af62f9f9ab Mon Sep 17 00:00:00 2001
From: Amr Khalifa <amrkhalifa@google.com>
Date: Mon, 5 Feb 2024 07:05:31 -0800
Subject: [PATCH] fixed math flex direction

PiperOrigin-RevId: 604306995
---
 docs/_static/css/custom.css | 3 +++
 docs/conf.py                | 6 +++++-
 2 files changed, 8 insertions(+), 1 deletion(-)
 create mode 100644 docs/_static/css/custom.css

diff --git a/docs/_static/css/custom.css b/docs/_static/css/custom.css
new file mode 100644
index 00000000..c71ea128
--- /dev/null
+++ b/docs/_static/css/custom.css
@@ -0,0 +1,3 @@
+div.math {
+  flex-direction: row;
+}
diff --git a/docs/conf.py b/docs/conf.py
index d312b5ef..adc13cf1 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -211,7 +211,11 @@ def new_process_docstring(app, what, name, obj, options, lines):
 # Add any paths that contain custom static files (such as style sheets) here,
 # relative to this directory. They are copied after the builtin static files,
 # so a file named "default.css" will overwrite the builtin "default.css".
-html_static_path = []
+
+html_static_path = ['_static']
+html_css_files = [
+    'css/custom.css',
+]
 
 # -- Options for myst -------------------------------------------------------
 nb_execution_mode = 'force'