@@ -3,3 +3,5 @@

 __version__ = "0.1.0.dev0"

 from qsim import state, gate, measurement

+

+__all__ = ["state", "gate", "measurement"]

@@ -38,7 +38,8 @@ def apply(op, qubits, state):

     # qubit corresponds to the *last* axis in the tensor etc.)

     qubit_axes = tuple(n_state_qubits - 1 - np.asarray(qubits))

-    # Applying the op to the state vector is then the tensor product over the appropriate axes

+    # Applying the op to the state vector is then the tensor product over the

+    # appropriate axes.

     axes = (np.arange(n_op_qubits, 2 * n_op_qubits), qubit_axes)

     new_state = np.tensordot(op, state, axes=axes)

@@ -68,7 +68,7 @@ def _check_valid_state(state):

         isinstance(state, np.ndarray)

         # is complex

         and np.issubdtype(state.dtype, np.complex128)

-        # is square

+        # is a vector

         and len(state.shape) == 1

         # has size 2**n, n > 1

         and np.log2(state.shape[0]).is_integer()