Make impurity public

FermiQ · Apr 7, 2021 · 17173fc · 17173fc
1 parent c14b0bf
commit 17173fc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tensorflow_quantum/python/differentiators/parameter_shift.py b/tensorflow_quantum/python/differentiators/parameter_shift.py
@@ -88,7 +88,7 @@ def get_gradient_circuits(self, programs, symbol_names, symbol_values):
         # Append impurity symbol into symbol name
         new_symbol_names = tf.concat([
             symbol_names,
-            tf.constant([parameter_shift_util._PARAMETER_IMPURITY_NAME])
+            tf.constant([parameter_shift_util.PARAMETER_IMPURITY_NAME])
         ], 0)
 
         # Symbol values are the input symbol values, tiled according to
@@ -201,7 +201,7 @@ def differentiate_analytic(self, programs, symbol_names, symbol_values,
         new_symbol_names = tf.concat([
             symbol_names,
             tf.expand_dims(tf.constant(
-                parameter_shift_util._PARAMETER_IMPURITY_NAME),
+                parameter_shift_util.PARAMETER_IMPURITY_NAME),
                            axis=0)
         ],
                                      axis=0)
@@ -347,7 +347,7 @@ def differentiate_sampled(self, programs, symbol_names, symbol_values,
         new_symbol_names = tf.concat([
             symbol_names,
             tf.expand_dims(tf.constant(
-                parameter_shift_util._PARAMETER_IMPURITY_NAME),
+                parameter_shift_util.PARAMETER_IMPURITY_NAME),
                            axis=0)
         ],
                                      axis=0)

diff --git a/tensorflow_quantum/python/differentiators/parameter_shift_util.py b/tensorflow_quantum/python/differentiators/parameter_shift_util.py
@@ -18,7 +18,7 @@
 
 from tensorflow_quantum.core.ops import tfq_ps_util_ops
 
-_PARAMETER_IMPURITY_NAME = '_param_shift'
+PARAMETER_IMPURITY_NAME = '_impurity_for_param_shift'
 
 
 @tf.function
@@ -65,7 +65,7 @@ def parse_programs(programs, symbol_names, symbol_values, n_symbols,
 
     # Collecting doped programs with impurity sympy.Symbol from all programs
     # with parameterized gates.
-    impurity = tf.tile(tf.convert_to_tensor([_PARAMETER_IMPURITY_NAME]),
+    impurity = tf.tile(tf.convert_to_tensor([PARAMETER_IMPURITY_NAME]),
                        [n_symbols])
     symbols = tf.convert_to_tensor(symbol_names)