Enable MathJax for Sphinx documentation to fix math rendering

Aniketsy · Aniketsy · commit f9be8f7a466a · 2025-09-14T11:14:35.000+05:30
diff --git a/docs/conf.py b/docs/conf.py
@@ -0,0 +1,3 @@
+extensions = [
+    'sphinx.ext.mathjax',
+]
diff --git a/tensorflow_quantum/core/ops/math_ops/fidelity_op.py b/tensorflow_quantum/core/ops/math_ops/fidelity_op.py
@@ -25,9 +25,11 @@ def fidelity(programs, symbol_names, symbol_values, other_programs):
     Compute (potentially many) fidelities between the given circuits and
     the symbol free comparison circuits.
 
-    Calculates out[i][j] = $ | \langle \psi_{\text{programs[i]}} \\
-     (\text{symbol\_values[i]}) | \psi_{\text{other\_programs[j]}} \rangle \\
-     |^2 $
+    Calculates out[i][j] as:
+    $$
+    |\langle \psi_{\text{programs[i]}}(\text{symbol\_values[i]}) \mid
+    \psi_{\text{other\_programs[j]}} \rangle|^2
+    $$
 
 
     >>> symbols = sympy.symbols('alpha beta')
diff --git a/tensorflow_quantum/core/ops/math_ops/inner_product_op.py b/tensorflow_quantum/core/ops/math_ops/inner_product_op.py
@@ -75,8 +75,11 @@ def inner_product(programs, symbol_names, symbol_values, other_programs):
     Compute (potentially many) inner products between the given circuits and
     the symbol free comparison circuits.
 
-    Calculates out[i][j] = $ \langle \psi_{\text{programs[i]}} \\
-     (\text{symbol\_values[i]}) | \psi_{\text{other\_programs[j]}} \rangle $
+    Calculates out[i][j] as:
+    $$
+    \langle \psi_{\text{programs[i]}}(\text{symbol\_values[i]}) \mid
+    \psi_{\text{other\_programs[j]}} \rangle
+    $$
 
 
     >>> symbols = sympy.symbols('alpha beta')

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+extensions = [`
	`2`	`+ 'sphinx.ext.mathjax',`
	`3`	`+]`