adapt to BlockArrays 1.1

ITensor · Jun 18, 2024 · fb59499 · fb59499
1 parent 43c8b4d
commit fb59499
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/NDTensors/Project.toml b/NDTensors/Project.toml
@@ -56,7 +56,7 @@ AMDGPU = "0.9"
 Accessors = "0.1.33"
 Adapt = "3.7, 4"
 ArrayLayouts = "1.4"
-BlockArrays = "1"
+BlockArrays = "1.1"
 CUDA = "5"
 Compat = "4.9"
 cuTENSOR = "2"

diff --git a/NDTensors/src/lib/GradedAxes/src/gradedunitrange.jl b/NDTensors/src/lib/GradedAxes/src/gradedunitrange.jl
@@ -42,10 +42,7 @@ end
 
 # == is just a range comparison that ignores labels. Need dedicated function to check equality.
 function gradedisequal(a1::AbstractUnitRange, a2::AbstractUnitRange)
-  # TODO remove workaround once BlockArrays.blockisequal is generalized to Integer
-  blocka1 = BlockArrays.blockedrange(GradedAxes.unlabel.(BlockArrays.blocklengths(a1)))
-  blocka2 = BlockArrays.blockedrange(GradedAxes.unlabel.(BlockArrays.blocklengths(a2)))
-  return blockisequal(blocka1, blocka2) && (blocklabels(a1) == blocklabels(a2))
+  return blockisequal(a1, a2) && (blocklabels(a1) == blocklabels(a2))
 end
 
 # TODO: Use `TypeParameterAccessors`.