Fix dependency bugs in stable/0.3 (#150)

DanPuzzuoli · web-flow · commit 6a1f5fdcf992 · 2022-11-15T20:14:46.000-05:00
diff --git a/qiskit_dynamics/signals/signals.py b/qiskit_dynamics/signals/signals.py
@@ -101,7 +101,7 @@ def __init__(
                 self._is_constant = True
 
             if envelope.backend == "jax":
-                self._envelope = lambda t: envelope * jnp.ones_like(t)
+                self._envelope = lambda t: envelope * jnp.ones_like(Array(t).data)
             else:
                 self._envelope = lambda t: envelope * np.ones_like(t)
         elif callable(envelope):
