Fix ParamResolver.param_dict type (#2550)

* fix type * change Basic to Symbol * change Symbol to Basic
quantumlib · Nov 14, 2019 · 23ce7ca · 23ce7ca
1 parent 4ff38cf
commit 23ce7ca
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/cirq/study/resolver.py b/cirq/study/resolver.py
@@ -22,7 +22,7 @@
     import cirq
 
 
-ParamDictType = Dict[Union[str, sympy.Basic], Union[float, str, sympy.Symbol]]
+ParamDictType = Dict[Union[str, sympy.Symbol], Union[float, str, sympy.Basic]]
 document(
     ParamDictType,  # type: ignore
     """Dictionary from symbols to values.""")
@@ -57,9 +57,8 @@ def __init__(self,
             return  # Already initialized. Got wrapped as part of the __new__.
 
         self._param_hash = None
-        self.param_dict = cast(
-            Dict[Union[str, sympy.Symbol], Union[float, str, sympy.Symbol]],
-            {} if param_dict is None else param_dict)
+        self.param_dict = cast(ParamDictType,
+                               {} if param_dict is None else param_dict)
 
     def value_of(self,
                  value: Union[sympy.Basic, float, str]) -> 'cirq.TParamVal':