Spaces:
Running
Running
Commit ·
2c18e0f
1
Parent(s): 4f499c4
fix: pad single-row input for SAP RPT-1 OSS predict (np.concatenate crash)
Browse files- webapp/benchmark.py +12 -1
webapp/benchmark.py
CHANGED
|
@@ -110,12 +110,23 @@ class _SAPModel:
|
|
| 110 |
return self
|
| 111 |
|
| 112 |
def predict(self, X):
|
| 113 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 114 |
if not self._real and self.task == "classification":
|
| 115 |
preds = self._le.inverse_transform(preds)
|
| 116 |
return preds
|
| 117 |
|
| 118 |
def predict_proba(self, X):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 119 |
return self._model.predict_proba(X)
|
| 120 |
|
| 121 |
@property
|
|
|
|
| 110 |
return self
|
| 111 |
|
| 112 |
def predict(self, X):
|
| 113 |
+
if self._real and len(X) == 1:
|
| 114 |
+
# SAP RPT-1 OSS crashes on single-row input (np.concatenate on 0-dim arrays).
|
| 115 |
+
# Duplicate the row to form a 2-row batch, then return only the first result.
|
| 116 |
+
import pandas as pd
|
| 117 |
+
X_pad = pd.concat([X, X], ignore_index=True) if hasattr(X, 'iloc') else np.vstack([X, X])
|
| 118 |
+
preds = self._model.predict(X_pad)[:1]
|
| 119 |
+
else:
|
| 120 |
+
preds = self._model.predict(X)
|
| 121 |
if not self._real and self.task == "classification":
|
| 122 |
preds = self._le.inverse_transform(preds)
|
| 123 |
return preds
|
| 124 |
|
| 125 |
def predict_proba(self, X):
|
| 126 |
+
if self._real and len(X) == 1:
|
| 127 |
+
import pandas as pd
|
| 128 |
+
X_pad = pd.concat([X, X], ignore_index=True) if hasattr(X, 'iloc') else np.vstack([X, X])
|
| 129 |
+
return self._model.predict_proba(X_pad)[:1]
|
| 130 |
return self._model.predict_proba(X)
|
| 131 |
|
| 132 |
@property
|