indexing fix

pablosanjose · Oct 20, 2024 · 51b499d · 51b499d
1 parent 6f99a50
commit 51b499d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/types.jl b/src/types.jl
@@ -559,10 +559,9 @@ orbranges(cs::Union{CellOrbitalsGrouped, CellOrbitalsGroupedDict}) = Iterators.a
 
 # range of orbital indices of dn cellorbs
 function orbrange(cs::AnyOrbitalSlice, dn::SVector)
-    for (dn´, rng) in zip(cells(cs), orbranges(cs))
-        dn == dn´ && return rng
-    end
-    return 0:-1
+    offset = get(offsets(cs), dn, 0)
+    rng = offset+1:offset+norbitals(cs, dn)
+    return rng
 end
 
 Base.isempty(s::LatticeSlice) = isempty(s.cellsdict)