* Code formatting

* Fix for observe

Signed-off-by: Pradnya Khalate <[email protected]>

Author: khalatepradnya

python/cudaq/runtime/sample.py

@@ -21,7 +21,9 @@ def __broadcastSample(kernel,
     N = len(argSet)
     results = []
     for i, a in enumerate(argSet):
-        ctx = cudaq_runtime.ExecutionContext("sample_explicit" if explicit_measurements else "sample", shots_count)
+        ctx = cudaq_runtime.ExecutionContext(
+            "sample_explicit" if explicit_measurements else "sample",
+            shots_count)
         ctx.totalIterations = N
         ctx.batchIteration = i
         cudaq_runtime.setExecutionContext(ctx)
@@ -110,7 +112,8 @@ def sample(kernel,
         cudaq_runtime.unset_noise()
         return res
 
-    ctx = cudaq_runtime.ExecutionContext("sample_explicit" if explicit_measurements else "sample", shots_count)
+    ctx = cudaq_runtime.ExecutionContext(
+        "sample_explicit" if explicit_measurements else "sample", shots_count)
     ctx.hasConditionalsOnMeasureResults = has_conditionals_on_measure_result
     cudaq_runtime.setExecutionContext(ctx)
 

runtime/nvqir/CircuitSimulator.h

@@ -1263,7 +1263,8 @@ class CircuitSimulatorBase : public CircuitSimulator {
   /// @brief Set the execution context
   void setExecutionContext(cudaq::ExecutionContext *context) override {
     executionContext = context;
-    currentSamplingMode = cudaq::details::getSamplingMode(context->name);
+    if (cudaq::details::isSamplingContext(executionContext))
+      currentSamplingMode = cudaq::details::getSamplingMode(context->name);
     executionContext->canHandleObserve = canHandleObserve();
     currentCircuitName = context->kernelName;
     CUDAQ_INFO("Setting current circuit name to {}", currentCircuitName);