From f78cb3ff27300c869b170238c56cfab0be8d64ce Mon Sep 17 00:00:00 2001 From: Ishtiaq Hussain <53497039+Ishticode@users.noreply.github.com> Date: Fri, 1 Sep 2023 23:09:52 +0100 Subject: [PATCH] refactor(frontend): remove get_n_split from StratifiedKFold and leave the call to the base class KFold in sklearn front --- ivy/functional/frontends/sklearn/model_selection/_split.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/ivy/functional/frontends/sklearn/model_selection/_split.py b/ivy/functional/frontends/sklearn/model_selection/_split.py index a298b48a1b91d..a5ca31b553b07 100644 --- a/ivy/functional/frontends/sklearn/model_selection/_split.py +++ b/ivy/functional/frontends/sklearn/model_selection/_split.py @@ -75,9 +75,6 @@ def __init__( def _iter_test_indices(self, X=None, y=None, groups=None): raise NotImplementedError - def get_n_splits(self, X=None, y=None, groups=None): - return self.n_splits - @to_ivy_arrays_and_back def train_test_split(