Update Source/DafnyRuntime/DafnyRuntimePython/_dafny/__init__.py

Co-authored-by: Fabio Madge <[email protected]>
dafny-lang · Jan 23, 2025 · b9b8b97 · b9b8b97
1 parent 2167e55
commit b9b8b97
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/Source/DafnyRuntime/DafnyRuntimePython/_dafny/__init__.py b/Source/DafnyRuntime/DafnyRuntimePython/_dafny/__init__.py
@@ -257,10 +257,8 @@ def __add__(self, other):
     def __getitem__(self, key):
         if isinstance(key, slice):
             start, stop, step = key.indices(len(self))
-            if isinstance(self.elems, _SeqSlice):
-                # Avoiding .Elements call on SeqSlice avoids creating a list, which is expensive
-                return Seq(_SeqSlice(self.elems, start=start, stop=stop, step=step), isStr=self.isStr)
-            return Seq(_SeqSlice(self.Elements, start=start, stop=stop, step=step), isStr=self.isStr)
+            elements = self.elems if isinstance(self.elems, _SeqSlice) else self.Elements
+            return Seq(_SeqSlice(elements, start=start, stop=stop, step=step), isStr=self.isStr)
         return self.Elements.__getitem__(key)
 
     def set(self, key, value):