From ffce8bf2d7617ce7f6765a5b06c33d4d9047ef58 Mon Sep 17 00:00:00 2001
From: Serge Panev <spanev@nvidia.com>
Date: Wed, 20 Nov 2019 20:07:23 +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..cb2153d4d94d 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: Invalid argument type, expecting %s, got %s"%(NDArray, type(x))
     hdl = SymbolHandle()
     check_call(_LIB.MXAutogradGetSymbol(x.handle, ctypes.byref(hdl)))
     return Symbol(hdl)