weird imports problem temporarily resolved

mfinzi · Feb 26, 2021 · e3b4ddf · e3b4ddf
1 parent b607c1f
commit e3b4ddf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/emlp/solver/groups.py b/emlp/solver/groups.py
@@ -11,7 +11,8 @@
 from jax import jit,vmap
 from functools import partial
 import logging
-#from .product_sum_reps import lazy_kron,lazy_kronsum,LazyPerm,I
+# this line needs to be here or imports break???
+from .product_sum_reps import rep_permutation 
 
 def rel_err(A,B):
     return jnp.mean(jnp.abs(A-B))/(jnp.mean(jnp.abs(A)) + jnp.mean(jnp.abs(B))+1e-6)