From 369d155a2d136e3eac065dd0ef6f28f702806ff2 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Thu, 28 Sep 2023 16:52:58 -0500
Subject: [PATCH] Einsum{Elementwise,Reduction}Axis: support ordering
---
pytato/array.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/pytato/array.py b/pytato/array.py
index 237d3be..9366dc8 100644
--- a/pytato/array.py
+++ b/pytato/array.py
@@ -922,7 +922,7 @@ class EinsumAxisDescriptor:
pass
-@attrs.frozen
+@attrs.frozen(order=True)
class EinsumElementwiseAxis(EinsumAxisDescriptor):
"""
Describes an elementwise access pattern of an array's axis. In terms of the
@@ -932,7 +932,7 @@ class EinsumElementwiseAxis(EinsumAxisDescriptor):
dim: int
-@attrs.frozen
+@attrs.frozen(order=True)
class EinsumReductionAxis(EinsumAxisDescriptor):
"""
Describes a reduction access pattern of an array's axis. In terms of the
--
GitLab