Fix type checking

CyberZHG · Sep 26, 2018 · a480f83 · a480f83
1 parent 35f7bb7
commit a480f83
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/keras_wc_embd/word_char_embd.py b/keras_wc_embd/word_char_embd.py
@@ -147,11 +147,11 @@ def get_embedding_layer(word_dict_len,
                 name='Dense_Char',
             ),
         ]
-    elif type(char_hidden_layer_type) is list or isinstance(char_hidden_layer_type, keras.layers.Layer):
+    elif isinstance(char_hidden_layer_type, list) or isinstance(char_hidden_layer_type, keras.layers.Layer):
         char_hidden_layer = char_hidden_layer_type
     else:
         raise NotImplementedError('Unknown character hidden layer type: %s' % char_hidden_layer_type)
-    if type(char_hidden_layer) is not list:
+    if not isinstance(char_hidden_layer, list):
         char_hidden_layer = [char_hidden_layer]
     for i, layer in enumerate(char_hidden_layer):
         if i == len(char_hidden_layer) - 1:

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='keras-word-char-embd',
-    version='0.14',
+    version='0.15',
     packages=['keras_wc_embd'],
     url='https://github.com/CyberZHG/keras-word-char-embd',
     license='MIT',