diff options
| author | Павел Жуков <33721692+LeaveMyYard@users.noreply.github.com> | 2023-05-11 17:06:23 +0300 |
|---|---|---|
| committer | Павел Жуков <33721692+LeaveMyYard@users.noreply.github.com> | 2023-05-11 17:06:23 +0300 |
| commit | a5deea864ae9622f20a376a42227aa21978f86af (patch) | |
| tree | b9cce29dc2508410f4c6d0521c922108a26b1e8a /robusta_krr/strategies | |
| parent | 717baa6b12e09b5103c05737c11b2f56dcf651ba (diff) | |
Fix calculation for Simple's CPU proposal
Diffstat (limited to 'robusta_krr/strategies')
| -rw-r--r-- | robusta_krr/strategies/simple.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/robusta_krr/strategies/simple.py b/robusta_krr/strategies/simple.py index 456123c..b8fa8d9 100644 --- a/robusta_krr/strategies/simple.py +++ b/robusta_krr/strategies/simple.py @@ -32,8 +32,12 @@ class SimpleStrategySettings(StrategySettings): if len(data) == 0: return float("NaN") - data_ = np.concatenate([values[:, 1] for values in data.values()]) if len(data) > 1 else list(data.values())[0] - return float(np.percentile(data_, self.cpu_percentile / 100)) + if len(data) > 1: + data_ = np.concatenate([values[:, 1] for values in data.values()]) + else: + data_ = list(data.values())[0][:, 1] + + return float(np.percentile(data_, self.cpu_percentile)) class SimpleStrategy(BaseStrategy[SimpleStrategySettings]): |