Skip to content

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

@lcwongme

Description

@lcwongme
  • 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):

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions