Skip to content

Commit c73d2d1

Browse files
hotfix: predictor path model loading (#123)
* fix: fallback loader for unknown yaml tags * fix: model loading for old models broken by predictor import * chore: bump to 0.23.1
1 parent 8f8077b commit c73d2d1

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

plexe/core/storage.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -230,6 +230,8 @@ def _load_model_data_from_tar(path: str | Path) -> Dict[str, Any]:
230230
predictor_source = None
231231
if "code/predictor.py" in [m.name for m in tar.getmembers()]:
232232
predictor_source = tar.extractfile("code/predictor.py").read().decode("utf-8")
233+
# FIXME: this is a hack required to ensure backwards compatibility with old models
234+
predictor_source = predictor_source.replace("plexe.internal.models.interfaces", "plexe.core.interfaces")
233235

234236
feature_transformer_source = None
235237
if "code/feature_transformer.py" in [m.name for m in tar.getmembers()]:

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "plexe"
3-
version = "0.23.0"
3+
version = "0.23.1"
44
description = "An agentic framework for building ML models from natural language"
55
authors = [
66
"marcellodebernardi <[email protected]>",

0 commit comments

Comments
 (0)