Enforces NDArray type in get_symbol (#16871)

Signed-off-by: Serge Panev <spanev@nvidia.com>
apache · Nov 20, 2019 · 95c3d03 · 95c3d03
1 parent 7c9cb6b
commit 95c3d03
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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)