Minor change to test.

Author: vbharadwaj-bk

tests/batch_test.py

@@ -317,7 +317,7 @@ def tp_and_problem(self, request, problem, extra_tp_constructor_args, with_jax):
 
             tp_base_cls = jax_tp
         else:
-            from openequivariance._torch.utils import transpose_irreps
+            from openequivariance import transpose_irreps
 
             tp_base_cls = oeq.TensorProduct