From 37b52b8d4c3510eb6c46f281c63846d8c3bd0f48 Mon Sep 17 00:00:00 2001
From: Junru Shao <junrushao1994@gmail.com>
Date: Tue, 26 Nov 2019 20:42:51 +0000
Subject: [PATCH] Allow Array/Map store objects that are not NodeRef

---
 include/tvm/node/container.h | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/include/tvm/node/container.h b/include/tvm/node/container.h
index c36c6c141451..41b47d3a679e 100644
--- a/include/tvm/node/container.h
+++ b/include/tvm/node/container.h
@@ -137,7 +137,7 @@ class IterAdapter {
  * \tparam T The content NodeRef type.
  */
 template<typename T,
-         typename = typename std::enable_if<std::is_base_of<NodeRef, T>::value>::type >
+         typename = typename std::enable_if<std::is_base_of<ObjectRef, T>::value>::type >
 class Array : public NodeRef {
  public:
   /*!
@@ -330,9 +330,9 @@ class Array : public NodeRef {
 template<typename K,
          typename V,
          typename = typename std::enable_if<
-           std::is_base_of<NodeRef, K>::value ||
+           std::is_base_of<ObjectRef, K>::value ||
            std::is_base_of<std::string, K>::value >::type,
-         typename = typename std::enable_if<std::is_base_of<NodeRef, V>::value>::type>
+         typename = typename std::enable_if<std::is_base_of<ObjectRef, V>::value>::type>
 class Map : public NodeRef {
  public:
   /*!