16598a0bb84e5980d3ac6f333f8249b5887fa2d0,heamy/dataset.py,Dataset,_setup_data,#Dataset#Any#Any#Any#Any#,146

Before Change

        self._X_train = X_train
        self._X_test = X_test

        if isinstance(y_train, (pd.Series, pd.DataFrame)):
            self._y_train = y_train.values
        else:
            self._y_train = y_train

        if isinstance(y_test, (pd.Series, pd.DataFrame)):
            self._y_test = y_test.values
        else:
            self._y_test = y_test

After Change

        self._X_test = self._validate_data(X_test)

        self._y_train = self._validate_data(y_train, only_numpy=True)
        self._y_test = self._validate_data(y_test, only_numpy=True)

    def _validate_data(self, data, only_numpy=False):
        if not only_numpy:
            dtypes = (pd.Series, pd.DataFrame, np.ndarray)

Instances