Add back missing prime

Author: mtfishman

ext/QuantumOperatorDefinitionsITensorBaseExt/QuantumOperatorDefinitionsITensorBaseExt.jl

@@ -22,4 +22,8 @@ function QuantumOperatorDefinitions.has_fermion_string(n::String, r::Index)
   return has_fermion_string(OpName(n), SiteType(r))
 end
 
+function Base.axes(::OpName, domain::Tuple{Vararg{Index}})
+  return (prime.(domain)..., dag.(domain)...)
+end
+
 end

src/op.jl

@@ -98,7 +98,7 @@ function array(a::AbstractArray, ax::Tuple{Vararg{AbstractUnitRange}})
   return a[ax...]
 end
 
-function state_or_op_axes(::OpName, domain::Tuple{Vararg{AbstractUnitRange}})
+function Base.axes(::OpName, domain::Tuple{Vararg{AbstractUnitRange}})
   return (domain..., domain...)
 end
 
@@ -108,7 +108,7 @@ function state_or_op_convert(
   domain::Tuple{Vararg{AbstractUnitRange}},
   a::AbstractArray,
 )
-  ax = state_or_op_axes(n, domain)
+  ax = axes(n, domain)
   a′ = reshape(a, length.(ax))
   a′′ = array(a′, ax)
   return convert(arrtype, a′′)

src/state.jl

@@ -32,7 +32,7 @@ macro state_alias(name1, name2, params...)
   return state_alias_expr(name1, name2)
 end
 
-function state_or_op_axes(::StateName, domain::Tuple{Vararg{AbstractUnitRange}})
+function Base.axes(::StateName, domain::Tuple{Vararg{AbstractUnitRange}})
   return domain
 end