better gitignore

.gitignore

@@ -138,4 +138,7 @@ dmypy.json
 
 # additional excludes specific to the repo
 *.t
-*.dat
+*.dat
+*.h5
+*.pkl
+*.json

emlp/solver/utils.py

@@ -9,6 +9,7 @@
 
 prod = lambda c: reduce(lambda a,b:a*b,c)
 
+
 # class TqdmToLogger(io.StringIO):
 #     """
 #         Output stream for TQDM which will output to logger module instead of

emlp/tests/equivariant_subspaces_tests.py

@@ -11,7 +11,7 @@
 import argparse
 import sys
 import copy
-from functools import partialmethod
+from functools import partialmethod,partial
 
 def rel_error(t1,t2):
     error = jnp.sqrt(jnp.mean(jnp.absolute(t1-t2)**2))