Use proper wrapper for indexing into module type data

apache · Feb 1, 2019 · e4ecb66 · e4ecb66
1 parent 06bc958
commit e4ecb66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/relay/pass/type_infer.cc b/src/relay/pass/type_infer.cc
@@ -461,7 +461,7 @@ class TypeInferencer : private ExprFunctor<Type(const Expr&)>,
     CHECK(mod_.defined())
       << "Cannot do type inference without a environment:"
       << c->name_hint;
-    TypeData td = mod_->type_definitions.at(c->belong_to);
+    TypeData td = mod_->LookupDef(c->belong_to);
     std::vector<Type> types;
     for (const auto & t : td->tv) {
       types.push_back(t);