apply changes from code review

PennyLaneAI · May 27, 2021 · ff90b90 · ff90b90
1 parent dc88464
commit ff90b90
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pennylane_cirq/simulator_device.py b/pennylane_cirq/simulator_device.py
@@ -118,9 +118,9 @@ def _expand_state(self, state_vector, wires):
         # get indices for which the state is changed to input state vector elements
         ravelled_indices = np.ravel_multi_index(unravelled_indices.T, [2] * self.num_wires)
 
-        state_vector = self._scatter(ravelled_indices, state_vector, [2 ** self.num_wires])
-        state_vector = self._reshape(state_vector, [2] * self.num_wires)
-        state_vector = self._asarray(state_vector, dtype=self.C_DTYPE)
+        state = np.zeros([2 ** self.num_wires], dtype=np.complex128)
+        state[ravelled_indices] = state_vector
+        state_vector = state.reshape([2] * self.num_wires)
 
         return state_vector.flatten()