Fix SAP RPT-1 single-row prediction and add hf_xet

matrix/benchmark.py

@@ -102,13 +102,43 @@ class _SAPModel:
         return self
 
     def predict(self, X):
-        preds = self._model.predict(X)
+        # Defensively handle single-row predictions which can crash some versions of sap_rpt_oss
+        is_single = len(X) == 1
+        if is_single:
+            if isinstance(X, pd.DataFrame):
+                X_in = pd.concat([X, X])
+            else:
+                X_in = np.vstack([X, X])
+        else:
+            X_in = X
+
+        preds = self._model.predict(X_in)
+        
+        if is_single:
+            # Take the first prediction only
+            if isinstance(preds, (list, np.ndarray)):
+                preds = preds[:1]
+
         if not self._real and self.task == "classification":
             preds = self._le.inverse_transform(preds)
         return preds
 
     def predict_proba(self, X):
-        return self._model.predict_proba(X)
+        is_single = len(X) == 1
+        if is_single:
+            if isinstance(X, pd.DataFrame):
+                X_in = pd.concat([X, X])
+            else:
+                X_in = np.vstack([X, X])
+        else:
+            X_in = X
+
+        proba = self._model.predict_proba(X_in)
+
+        if is_single:
+            if isinstance(proba, (list, np.ndarray)):
+                proba = proba[:1]
+        return proba
 
     @property
     def label(self):

requirements.txt

@@ -4,6 +4,8 @@ uvicorn[standard]>=0.29.0
 python-multipart>=0.0.9
 python-dotenv>=1.0.0
 jinja2>=3.1.3
+hf_xet
+huggingface_hub[hf_xet]
 
 # Data & Science Stack
 numpy==1.26.4