Allow build layer to recognize layers imported from tensorflow keras - Githubissues

MLBazaar / MLPrimitives

Primitives for machine learning and data science.

https://mlbazaar.github.io/MLPrimitives

MIT License

69 stars 38 forks source link

Allow build layer to recognize layers imported from tensorflow keras #268

Open lcwong0928 opened 2 years ago

lcwong0928 commented 2 years ago

MLPrimitives version: 0.3.3.dev0
Python version: 3.7.0
Operating System: macOS

Description

Allow ml.primitives.adapters.build_layer to also recognize layers imported from tensorflow.keras.

What I Did

Current Version:

if issubclass(layer_class, keras.layers.wrappers.Wrapper):

Suggested Changes:

if issubclass(layer_class, tf.keras.layers.Wrapper) or issubclass(layer_class, keras.layers.wrappers.Wrapper):