From eed23551fd99ee10acd63566dedc89ec32e66fc8 Mon Sep 17 00:00:00 2001
From: Serge Panev <spanev@nvidia.com>
Date: Wed, 20 Nov 2019 14:42:40 +0100
Subject: [PATCH] Enforces NDArray type in get_symbol

Signed-off-by: Serge Panev <spanev@nvidia.com>
---
 python/mxnet/autograd.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/mxnet/autograd.py b/python/mxnet/autograd.py
index 6f1cc4367821..f10b5ce3ed6d 100644
--- a/python/mxnet/autograd.py
+++ b/python/mxnet/autograd.py
@@ -360,6 +360,8 @@ def get_symbol(x):
     Symbol
         The retrieved Symbol.
     """
+    assert isinstance(x, NDArray), \
+       "get_symbol: Expecting %s, got %s"%(NDArray, type(x))
     hdl = SymbolHandle()
     check_call(_LIB.MXAutogradGetSymbol(x.handle, ctypes.byref(hdl)))
     return Symbol(hdl)