remove vec

ITensor · Nov 14, 2024 · 1756c3c · 1756c3c
1 parent 60adbee
commit 1756c3c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/NDTensors/src/lib/GradedAxes/src/gradedunitrangedual.jl b/NDTensors/src/lib/GradedAxes/src/gradedunitrangedual.jl
@@ -91,8 +91,7 @@ end
 function dual_axes(v::BlockVector)
   # dual both v elements and v axes
   block_axes = dual.(axes(v))
-  flipped = mortar(dual.(vec.(blocks(v))), block_axes)
-  return flipped
+  return mortar(dual.(blocks(v)), block_axes)
 end
 
 Base.axes(a::GradedUnitRangeDual) = axes(nondual(a))

diff --git a/NDTensors/src/lib/GradedAxes/test/test_dual.jl b/NDTensors/src/lib/GradedAxes/test/test_dual.jl
@@ -219,7 +219,7 @@ end
     @test label(ad[Block(2)]) == U1(-1)
     @test label(ad[Block(2)[1:1]]) == U1(-1)
 
-    v = ad[mortar([Block(2)[1:1]])]
+    v = ad[[Block(2)[1:1]]]
     @test v isa AbstractVector{LabelledInteger{Int64,U1}}
     @test length(v) == 1
     @test label(first(v)) == U1(-1)
@@ -228,16 +228,16 @@ end
     @test isdual(axes(v, 1))
     @test blocklabels(axes(v, 1)) == [U1(-1)]
 
-    v = ad[[Block(2)]]
+    v = ad[mortar([Block(2)[1:1]])]
     @test v isa AbstractVector{LabelledInteger{Int64,U1}}
-    @test isdual(axes(v, 1))  # used in view(::BlockSparseVector, [Block(1)])
+    @test isdual(axes(v, 1))  # used in view(::BlockSparseVector, [Block(1)[1:1]])
     @test label(first(v)) == U1(-1)
     @test unlabel(first(v)) == 3
     @test blocklabels(axes(v, 1)) == [U1(-1)]
 
-    v = ad[mortar([Block(2)[1:1]])]
+    v = ad[[Block(2)]]
     @test v isa AbstractVector{LabelledInteger{Int64,U1}}
-    @test isdual(axes(v, 1))  # used in view(::BlockSparseVector, [Block(1)[1:1]])
+    @test isdual(axes(v, 1))  # used in view(::BlockSparseVector, [Block(1)])
     @test label(first(v)) == U1(-1)
     @test unlabel(first(v)) == 3
     @test blocklabels(axes(v, 1)) == [U1(-1)]