Update src/main/scala/util/HeterogeneousBag.scala

Co-authored-by: Jack Koenig <koenig@sifive.com>
chipsalliance · May 10, 2023 · b969099 · b969099
1 parent 9a1d783
commit b969099
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/scala/util/HeterogeneousBag.scala b/src/main/scala/util/HeterogeneousBag.scala
@@ -9,7 +9,10 @@ import scala.collection.immutable.VectorMap
 final case class HeterogeneousBag[T <: Data](elts: Seq[T]) extends Record with collection.IndexedSeq[T] {
   val elements = VectorMap(elts.zipWithIndex.map { case (n,i) => (i.toString, DataMirror.internal.chiselTypeClone(n)) }:_*)
 
-  def apply(x: Int): T = elements.values.toSeq(x)
+  def apply(x: Int): T = {
+    val key = elements.keys(x) // VectorMap.keys returns a Vector
+    elements(key)
+  }
   def length = elts.length
 
   override def className: String = super.className