Reformatted tests with new black version

tests/test_ops.py

@@ -1,6 +1,7 @@
 """
 Unit tests for GPI2 operation and its utility functions.
 """
+
 import pytest
 import torch
 import jax