From 5b30977d0f22e9c80b33c2913f11a546b2cc88b0 Mon Sep 17 00:00:00 2001
From: Erick Ochoa Lopez <erick.ochoalopez@xanadu.ai>
Date: Wed, 31 Jul 2024 19:39:00 -0400
Subject: [PATCH] clang format

---
 frontend/catalyst/jax_tracer.py                     | 2 ++
 frontend/catalyst/third_party/oqc/src/OQCDevice.cpp | 5 ++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/frontend/catalyst/jax_tracer.py b/frontend/catalyst/jax_tracer.py
index 105d83bc86..114db89522 100644
--- a/frontend/catalyst/jax_tracer.py
+++ b/frontend/catalyst/jax_tracer.py
@@ -286,6 +286,8 @@ def _shapes(ts):
     if all_ends_with_qreg:  # [1]
         length = types[-1][-1].length
         types = [t[:-1] for t in types]
+    else:
+        length = None
     results = list(map(partial(reduce, jnp.promote_types), zip(*types)))
     return results + ([AbstractQreg(length)] if all_ends_with_qreg else [])
 
diff --git a/frontend/catalyst/third_party/oqc/src/OQCDevice.cpp b/frontend/catalyst/third_party/oqc/src/OQCDevice.cpp
index 15ecef2217..60587f1536 100644
--- a/frontend/catalyst/third_party/oqc/src/OQCDevice.cpp
+++ b/frontend/catalyst/third_party/oqc/src/OQCDevice.cpp
@@ -104,7 +104,10 @@ void OQCDevice::PartialCounts(DataView<double, 1> &eigvals, DataView<int64_t, 1>
 // After this poitn everything is unsupported
 auto OQCDevice::AllocateQubit() -> QubitIdType { RT_FAIL("Unsupported functionality"); }
 void OQCDevice::PrintState() { RT_FAIL("Unsupported functionality"); }
-void OQCDevice::SetState(std::vector<std::complex<double>>) { RT_FAIL("Unsupported functionality"); }
+void OQCDevice::SetState(std::vector<std::complex<double>>)
+{
+    RT_FAIL("Unsupported functionality");
+}
 
 void OQCDevice::Counts(DataView<double, 1> &eigvals, DataView<int64_t, 1> &counts, size_t shots)
 {