Adding a check for the vector size and formatting

Signed-off-by: Sachin Pisal <[email protected]>

python/utils/OpaqueArguments.h

@@ -99,6 +99,14 @@ inline py::args simplifiedValidateInputArguments(py::args &args) {
       if (shape.size() != 1)
         throw std::runtime_error("Cannot pass ndarray with shape != (N,).");
 
+      // Vector size validation
+      size_t size = shape[0].cast<size_t>();
+      if (size == 0 || (size & (size - 1)) != 0) {
+        throw std::invalid_argument(fmt::format(
+            "Invalid vector size: {}. Vector size must be a power of 2.",
+            size));
+      }
+
       arg = args[i].attr("tolist")();
     } else if (py::isinstance<py::str>(arg)) {
       arg = py::cast<std::string>(arg);

runtime/cudaq/algorithms/observe.h

@@ -59,9 +59,9 @@ concept ObserveCallValid =
 /// @param shots number of shots to run for the given kernel, or -1 if not
 /// applicable.
 /// @param noise noise model to use for the sample operation
-/// @param num_trajectories is the optional number of trajectories to be used when
-/// computing the expectation values in the presence of noise. This parameter is
-/// only applied to simulation backends that support noisy
+/// @param num_trajectories is the optional number of trajectories to be used
+/// when computing the expectation values in the presence of noise. This
+/// parameter is only applied to simulation backends that support noisy
 /// simulation of trajectories.
 struct observe_options {
   int shots = -1;