From 01ee5cc8f28ff34de360a783837739277a1298fd Mon Sep 17 00:00:00 2001 From: Snomaan6846 Date: Mon, 9 Feb 2026 12:36:36 +0530 Subject: [PATCH] fix: XGBoost model loading issue with modelcar Signed-off-by: Snomaan6846 rh-pre-commit.version: 2.3.2 rh-pre-commit.check-secrets: ENABLED --- runtimes/xgboost/mlserver_xgboost/xgboost.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/runtimes/xgboost/mlserver_xgboost/xgboost.py b/runtimes/xgboost/mlserver_xgboost/xgboost.py index 9e97fe132..aa992faaa 100644 --- a/runtimes/xgboost/mlserver_xgboost/xgboost.py +++ b/runtimes/xgboost/mlserver_xgboost/xgboost.py @@ -22,15 +22,18 @@ def _load_sklearn_interface(model_uri: str) -> XGBModel: + with open(model_uri, "rb") as f: + model_bytes = bytearray(f.read()) + try: regressor = xgb.XGBRegressor() - regressor.load_model(model_uri) + regressor.load_model(model_bytes) return regressor except TypeError: # If there was an error, it's likely due to the model being a # classifier classifier = xgb.XGBClassifier() - classifier.load_model(model_uri) + classifier.load_model(model_bytes) return classifier