Serializable Issue K8s Word Embeddings

[danilojsl]

Description

Extending serializable class to make correctly transfer data to executors

Motivation and Context

The following error is rasing on K8s setups:

NotSerializableException: com.johnsnowlabs.storage.BytesKey
Serialization stack:
	- object not serializable (class: com.johnsnowlabs.storage.BytesKey, value: com.johnsnowlabs.storage.BytesKey@49af0f2)
	- writeObject data (class: scala.collection.immutable.HashMap$SerializationProxy)
	- object (class scala.collection.immutable.HashMap$SerializationProxy, scala.collection.immutable.HashMap$SerializationProxy@a0fa7f5)
	- writeReplace data (class: scala.collection.immutable.HashMap$SerializationProxy)
	- object (class scala.collection.immutable.HashMap$HashTrieMap, Map(com.johnsnowlabs.storage.BytesKey@49af0f2 -> [B@299e1b8f, com.johnsnowlabs.storage.BytesKey@7659bba -> [B@6f512ea0, com.johnsnowlabs.storage.BytesKey@6cb12ff4 -> [B@21e28f54, com.johnsnowlabs.storage.BytesKey@41a114 -> [B@55d56360, com.johnsnowlabs.storage.BytesKey@d3312d20 -> [B@3330d7c4, com.johnsnowlabs.storage.BytesKey@838ea3e -> [B@6b0c9c32, com.johnsnowlabs.storage.BytesKey@28467149 -> [B@23041fc7, com.johnsnowlabs.storage.BytesKey@5f1a7784 -> [B@8892e4c, com.johnsnowlabs.storage.BytesKey@ebeb7311 -> [B@7c17f89a, com.johnsnowlabs.storage.BytesKey@3f528a -> [B@72867e92, com.johnsnowlabs.storage.BytesKey@16cf07f7 -> [B@45361fc8, com.johnsnowlabs.storage.BytesKey@4a3fe5fa -> [B@6d3b9dbf, com.johnsnowlabs.storage.BytesKey@71c268a5 -> [

How Has This Been Tested?

Screenshots (if appropriate):

Testing is still a WIP

Types of changes

• Bug fix (non-breaking change which fixes an issue)
• Code improvements with no or little impact
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to change)

Checklist:

• My code follows the code style of this project.
• My change requires a change to the documentation.
• I have updated the documentation accordingly.
• I have read the CONTRIBUTING page.
• I have added tests to cover my changes.
• All new and existing tests passed.