[query] Fix type serialization codegen (#12424)

hail-is · Nov 28, 2022 · e0f7e15 · e0f7e15
1 parent 00ee515
commit e0f7e15
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/hail/src/main/scala/is/hail/expr/ir/EmitClassBuilder.scala b/hail/src/main/scala/is/hail/expr/ir/EmitClassBuilder.scala
@@ -470,8 +470,8 @@ class EmitClassBuilder[C](
       Code.checkcast[T](
         Code.invokeScalaObject1[String, PType](
           IRParser.getClass, "parsePType", t.toString)))
-    pTypeMap.getOrElseUpdate(t,
-      genLazyFieldThisRef[T](setup)).get.asInstanceOf[Code[T]]
+    Code.checkcast[T](pTypeMap.getOrElseUpdate(t,
+      genLazyFieldThisRef[T](setup)).get)
   }
 
   def getType[T <: Type : TypeInfo](t: T): Code[T] = {
@@ -480,8 +480,8 @@ class EmitClassBuilder[C](
       Code.checkcast[T](
         Code.invokeScalaObject1[String, Type](
           IRParser.getClass, "parseType", t.parsableString())))
-    typMap.getOrElseUpdate(t,
-      genLazyFieldThisRef[T](setup)).get.asInstanceOf[Code[T]]
+    Code.checkcast[T](typMap.getOrElseUpdate(t,
+      genLazyFieldThisRef[T](setup)).get)
   }
 
   def getOrdering(t1: SType,