Coverage for qml_essentials / utils.py: 100%

1import jax 

2 

3# Re-exported for backwards compatibility: the Pauli-Clifford circuit transform 

4# now lives in :mod:`qml_essentials.pauli` (integrated with the symbolic 

5# :class:`~qml_essentials.operations.PauliWord` core). 

6from qml_essentials.pauli import PauliCircuit # noqa: F401 

7 

8 

9def safe_random_split(random_key: jax.random.PRNGKey, *args, **kwargs): 

10 if random_key is None: 

11 return None, None 

12 else: 

13 return jax.random.split(random_key, *args, **kwargs)