Akshay4506 commited on
Commit
2c18e0f
·
1 Parent(s): 4f499c4

fix: pad single-row input for SAP RPT-1 OSS predict (np.concatenate crash)

Browse files
Files changed (1) hide show
  1. 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
- preds = self._model.predict(X)
 
 
 
 
 
 
 
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