Remove _get_dims_length

Author: albertomercurio

src/qobj/energy_restricted.jl

@@ -70,6 +70,8 @@ function Base.show(io::IO, s::EnrSpace)
     return nothing
 end
 
+Base.length(s::EnrSpace{N}) where {N} = N
+
 Base.:(==)(s_enr1::EnrSpace, s_enr2::EnrSpace) = (s_enr1.size == s_enr2.size) && (s_enr1.dims == s_enr2.dims)
 
 dimensions_to_dims(s_enr::EnrSpace) = s_enr.dims

src/qobj/quantum_object_base.jl

@@ -240,9 +240,7 @@ get_dimensions_from(
 
 # this creates a list of HilbertSpace(1), it is used to generate `from` for Ket, and `to` for Bra
 space_one_list(dimensions::NTuple{N,AbstractSpace}) where {N} =
-    ntuple(i -> HilbertSpace(1), Val(sum(_get_dims_length, dimensions)))
-_get_dims_length(::HilbertSpace) = 1
-_get_dims_length(::EnrSpace{N}) where {N} = N
+    ntuple(i -> HilbertSpace(1), Val(sum(length, dimensions)))
 
 # functions for getting Float or Complex element type
 _float_type(A::AbstractQuantumObject) = _float_type(eltype(A))

src/qobj/space.jl

@@ -22,4 +22,6 @@ struct HilbertSpace <: AbstractSpace
     end
 end
 
+Base.length(s::HilbertSpace) = 1
+
 dimensions_to_dims(s::HilbertSpace) = SVector{1,Int}(s.size)