Fix linter bugs

ArgonneCPAC · Feb 9, 2024 · 7489f4d · 7489f4d
1 parent 701e1b1
commit 7489f4d
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/diffmah/rockstar_pdf_model.py b/diffmah/rockstar_pdf_model.py
@@ -3,7 +3,6 @@
 from collections import OrderedDict
 
 from jax import jit as jjit
-from jax import lax
 from jax import numpy as jnp
 from jax import vmap
 

diff --git a/diffmah/tng_pdf_model.py b/diffmah/tng_pdf_model.py
@@ -4,7 +4,6 @@
 from collections import OrderedDict
 
 from jax import jit as jjit
-from jax import lax
 from jax import numpy as jnp
 from jax import vmap