Fix keras modules

CyberZHG · Jun 20, 2021 · 02801c3 · 02801c3
1 parent 0c9c0ce
commit 02801c3
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 8 deletions.
diff --git a/keras_wc_embd/__init__.py b/keras_wc_embd/__init__.py
@@ -1,4 +1,4 @@
 from .word_char_embd import *
 from .wrapper import WordCharEmbd
 
-__version__ = '0.21.0'
+__version__ = '0.22.0'
diff --git a/keras_wc_embd/backend.py b/keras_wc_embd/backend.py
@@ -2,8 +2,8 @@
 from distutils.util import strtobool
 
 __all__ = [
-    'keras', 'utils', 'activations', 'applications', 'backend', 'datasets',
-    'layers', 'preprocessing', 'wrappers', 'callbacks', 'constraints', 'initializers',
+    'keras', 'utils', 'activations', 'backend',
+    'layers', 'callbacks', 'constraints', 'initializers',
     'metrics', 'models', 'losses', 'optimizers', 'regularizers', 'TF_KERAS',
 ]
 
@@ -17,12 +17,8 @@
 
 utils = keras.utils
 activations = keras.activations
-applications = keras.applications
 backend = keras.backend
-datasets = keras.datasets
 layers = keras.layers
-preprocessing = keras.preprocessing
-wrappers = keras.wrappers
 callbacks = keras.callbacks
 constraints = keras.constraints
 initializers = keras.initializers

diff --git a/tests/test_get_embedding_layer.py b/tests/test_get_embedding_layer.py
@@ -1,6 +1,6 @@
 import unittest
 import numpy
-import keras
+from keras_wc_embd.backend import keras
 from keras_wc_embd import get_embedding_layer