ebonite.core.objects.wrapper module¶

class ebonite.core.objects.wrapper.ModelIO[source]¶

Bases: ebonite.core.objects.wrapper.ModelIO, pyjackson.decorators.SubtypeRegisterMixin

Helps model wrapper with IO

Must be pyjackson-serializable

class ebonite.core.objects.wrapper.ModelWrapper(io: ebonite.core.objects.wrapper.ModelIO)[source]¶

Bases: ebonite.core.objects.wrapper.ModelWrapper, pyjackson.decorators.SubtypeRegisterMixin

Base class for model wrapper. Wrapper is an object that can save, load and inference a model

Must be pyjackson-serializable

class ebonite.core.objects.wrapper.LibModelWrapperMixin(io: ebonite.core.objects.wrapper.ModelIO)[source]¶

ModelWrapper mixin which provides model object requirements list consisting of PIP packages represented by module objects in libraries field.

class ebonite.core.objects.wrapper.WrapperArtifactCollection(wrapper: ebonite.core.objects.wrapper.ModelWrapper)[source]¶

This is a proxy ArtifactCollection for not persisted artifacts. Internally uses dump() to create model artifacts

Parameters:	wrapper – `ModelWrapper` instance

materialize(path)[source]¶

Calls dump() to materialize model in path

Parameters:	path – path to materialize model

bytes_dict() → Dict[str, bytes][source]¶: Calls dump() to get model artifacts bytes dict :return: dict artifact name -> bytes

blob_dict() → Iterable[Dict[str, ebonite.core.objects.artifacts.Blob]][source]¶

Calls dump() to get model artifacts blob dict

Returns:	dict artifact name -> `Blob`

class ebonite.core.objects.wrapper.PickleModelIO[source]¶

ModelIO for pickle-able models

When model is dumped, recursively checks objects if they can be dumped with ModelIO instead of pickling

So, if you use function that internally calls tensorflow model, this tensorflow model will be dumped with tensorflow code and not pickled

dump(model) → ebonite.core.objects.artifacts.ArtifactCollection[source]¶

Dumps model artifacts as ArtifactCollection

Returns:	context manager with `ArtifactCollection`

load(path)[source]¶

Loads artifacts into model field

Parameters:	path – path to load from

class ebonite.core.objects.wrapper.CallableMethodModelWrapper[source]¶

ModelWrapper implementation for functions