Fix numba.types shim

dmlc · Aug 27, 2018 · 8de29ec · 8de29ec
1 parent e7ca34c
commit 8de29ec
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/gluonnlp/base.py b/gluonnlp/base.py
@@ -42,9 +42,18 @@ def numba_jitclass(spec):
         # pylint: disable=unused-argument
         return identity
 
-    def numba_types(*args, **kwargs):
-        # pylint: disable=unused-argument
-        return None
+    class NumbaTypes(object):
+        class NumbaType(object):
+            def __getitem__(self, x):
+                pass
+
+            def __call__(self, *args, **kwargs):
+                # pylint: disable=unused-argument
+                pass
+
+        def __getattr__(self, attr):
+            return self.NumbaType()
 
+    numba_types = NumbaTypes()
     numba_njit = identity
     numba_prange = range