Merge branch 'warp' of github.com:atulcthakur/matgl_local into warp

Author: atulcthakur

src/matgl/kernels/compose_tensor.py

@@ -27,6 +27,8 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 """Warp kernels for composing 3x3 tensors from I, A, S components."""
 
+from __future__ import annotations
+
 import warp as wp
 
 from .utils import add_module, get_wp_fp_dtype

src/matgl/kernels/decompose_tensor.py

@@ -27,6 +27,8 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 """Warp kernels for decomposing 3x3 tensors into I, A, S components."""
 
+from __future__ import annotations
+
 import warp as wp
 
 from .utils import add_module, get_wp_fp_dtype

src/matgl/kernels/equivariant_o3_matmul.py

@@ -27,6 +27,8 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 """Warp kernels for O(3)-equivariant 3x3 tensor matrix multiplication."""
 
+from __future__ import annotations
+
 import warp as wp
 
 from .utils import add_module, get_wp_fp_dtype

src/matgl/kernels/equivariant_so3_matmul.py

@@ -27,6 +27,8 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 """Warp kernels for SO(3)-equivariant 3x3 tensor matrix multiplication."""
 
+from __future__ import annotations
+
 import warp as wp
 
 from .utils import add_module, get_wp_fp_dtype

src/matgl/kernels/graph_transform.py

@@ -27,6 +27,8 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 """Warp kernels for graph edge index transformation to sparse CSR format."""
 
+from __future__ import annotations
+
 import warp as wp
 
 

src/matgl/kernels/tensor_norm3.py

@@ -27,6 +27,8 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 """Warp kernels for computing 3x3 tensor norms (I, A, S components)."""
 
+from __future__ import annotations
+
 import warp as wp
 
 from .utils import add_module, get_wp_fp_dtype

src/matgl/kernels/tensornet_mp.py

@@ -25,6 +25,8 @@
 # CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+from __future__ import annotations
+
 import warp as wp
 
 from .utils import add_module, get_wp_fp_dtype

src/matgl/kernels/tensornet_radial_mp.py

@@ -25,6 +25,7 @@
 # CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+from __future__ import annotations
 
 import warp as wp