Add checks for numpy

import might want to go into _getshapeof

Author: Tadaboody

src/jupyter_contrib_nbextensions/nbextensions/varInspector/main.js

@@ -130,21 +130,20 @@ function html_table(jsonVars) {
     var varList = JSON.parse(String(jsonVars))
 
     var shape_str;
-    if(kernelLanguage === "python")
-    {
+    var has_shape = false;
+    if (varList.some(list_var => "varShape" in list_var )) { //if any of them have a shape
         shape_str = '<th >Shape</th>';
+        has_shape = true;
     }
     var beg_table = '<div class=\"inspector\"><table class=\"table fixed table-condensed table-nonfluid \"><col /> \
  <col  /><col /><thead><tr><th >X</th><th >Name</th><th >Type</th><th >Size</th>' + shape_str + '<th >Value</th></tr></thead><tr><td> \
  </td></tr>'
     var nb_vars = varList.length;
     var shape_col_str;
     for (var i = 0; i < nb_vars; i++) {
-        if(kernelLanguage === "python")
-        {
+        if (has_shape) {
             shape_col_str = '</td><td>' + varList[i].varShape + '</td><td>';
-        }else
-        {
+        } else {
             shape_col_str = '';
         }
         beg_table = beg_table +

src/jupyter_contrib_nbextensions/nbextensions/varInspector/var_list.py

@@ -3,11 +3,16 @@
 
 from IPython import get_ipython
 from IPython.core.magics.namespace import NamespaceMagics
-import numpy as np
 _nms = NamespaceMagics()
 _Jupyter = get_ipython()
 _nms.shell = _Jupyter.kernel.shell
 
+has_numpy = False
+try:
+    import numpy as np
+    has_numpy = True
+except ImportError:
+    has_numpy = False
 
 def _getsizeof(x):
     # return the size of variable x. Amended version of sys.getsizeof
@@ -22,16 +27,21 @@ def _getsizeof(x):
 def _getshapeof(x):
     #returns the shape of x if it has one
     #returns None otherwise - might want to return an empty string for an empty collum
-    try:
-        return x.shape
-    except AttributeError: #x does not have a shape
-        return None
+    global has_numpy
+    if has_numpy:
+        try:
+            return x.shape
+        except AttributeError: #x does not have a shape
+            return None
 
 def var_dic_list():
     types_to_exclude = ['module', 'function', 'builtin_function_or_method',
                         'instance', '_Feature', 'type', 'ufunc']
     values = _nms.who_ls()
-    vardic = [{'varName': v, 'varType': type(eval(v)).__name__, 'varSize': str(_getsizeof(eval(v))), 'varShape': str(_getshapeof(eval(v))), 'varContent': str(eval(v))[:200]}  # noqa
+    vardic = [{'varName': v, 'varType': type(eval(v)).__name__, 'varSize': str(_getsizeof(eval(v))), 'varContent': str(eval(v))[:200]}  # noqa
+    if has_numpy:
+        vardic['varSize'] = str(eval(_getshapeof(v)))
+    
     for v in values if (v not in ['_html', '_nms', 'NamespaceMagics', '_Jupyter']) & (type(eval(v)).__name__ not in types_to_exclude)] # noqa 
     return json.dumps(vardic)