use TensorAlgebra.trivial_axis

Author: ogauthe

Project.toml

@@ -21,7 +21,7 @@ WignerSymbols = "9f57e263-0b3d-5e2e-b1be-24f2bb48858b"
 Accessors = "0.1.42"
 BlockArrays = "1.6"
 BlockSparseArrays = "0.7.4"
-GradedArrays = "0.4.7"
+GradedArrays = "0.4.13"
 HalfIntegers = "1.6"
 LRUCache = "1.6"
 LinearAlgebra = "1.10"

src/fusiontensor/fusiontensor.jl

@@ -19,7 +19,7 @@ using GradedArrays:
   sectormergesort,
   sectors,
   space_isequal
-using TensorAlgebra: BlockedTuple, tuplemortar
+using TensorAlgebra: BlockedTuple, trivial_axis, tuplemortar
 using TensorProducts: tensor_product
 using TypeParameterAccessors: type_parameters
 
@@ -56,7 +56,7 @@ function flip_domain(nonflipped_col_axis, nonflipped_trees_to_ranges)
 end
 
 function fuse_axes(::Type{S}, ::Tuple{}) where {S<:AbstractSector}
-  fused_axis = dummy_axis(S)
+  fused_axis = trivial_axis(S)
   trees_to_ranges_mapping = Dict([SectorFusionTree{S}() => Block(1)[1:1]])
   return fused_axis, trees_to_ranges_mapping
 end

src/fusiontensor/fusiontensoraxes.jl

@@ -7,17 +7,12 @@ using GradedArrays:
   dual,
   sector_type,
   trivial
-using TensorAlgebra: AbstractBlockPermutation, BlockedTuple
+using TensorAlgebra: TensorAlgebra, AbstractBlockPermutation, BlockedTuple
 using TensorProducts: ⊗
 using TypeParameterAccessors: type_parameters
 
 # =======================================  Misc  ===========================================
 
-dummy_axis() = dummy_axis(TrivialSector)
-dummy_axis(x) = dummy_axis(typeof(x))
-dummy_axis(::Type{T}) where {T} = dummy_axis(sector_type(T))
-dummy_axis(::Type{S}) where {S<:AbstractSector} = gradedrange([trivial(S) => 1])
-
 promote_sector_type(legs::Tuple) = promote_sector_type(legs...)
 
 function promote_sector_type(legs...)
@@ -113,6 +108,10 @@ function GradedArrays.sector_type(::Type{FTA}) where {BT,FTA<:FusionTensorAxes{B
   return sector_type(type_parameters(type_parameters(BT, 3), 1))
 end
 
+# ==============================  TensorAlgebra interface  =================================
+
+TensorAlgebra.trivial_axis(fta::FusionTensorAxes) = trivial_axis(sector_type(fta))
+
 # ==============================  FusionTensor interface  ==================================
 
 codomain_axes(fta::FusionTensorAxes) = fta[Block(1)]
@@ -121,14 +120,14 @@ domain_axes(fta::FusionTensorAxes) = fta[Block(2)]
 
 function codomain_axis(fta::FusionTensorAxes)
   if ndims_codomain(fta) == 0
-    return dummy_axis(fta)
+    return trivial_axis(fta)
   end
   return ⊗(codomain_axes(fta)...)
 end
 
 function domain_axis(fta::FusionTensorAxes)
   if ndims_domain(fta) == 0
-    return dual(dummy_axis(fta))
+    return dual(trivial_axis(fta))
   end
   return dual(⊗(dual.(domain_axes(fta))...))
 end

test/test_basics.jl

@@ -1,5 +1,6 @@
 using Test: @test, @test_throws, @testset
 
+using BlockArrays: Block
 using BlockSparseArrays: BlockSparseArray
 using FusionTensors:
   FusionTensor,