Как я могу построить дерево решений с глубиной 0, но параметр "max_leaf_nodes = 1 должен иметь значение None или больше 1"?

Question

Я пытаюсь построить дерево решений с 1 корневым узлом, а именно.

import pandas as pd
import numpy as np
from matplotlib import pyplot as plt
from sklearn.model_selection import train_test_split, GridSearchCV
from sklearn.metrics import accuracy_score
from sklearn.tree import DecisionTreeClassifier, export_graphviz, DecisionTreeRegressor

X = np.linspace(-2, 2, 7)
y = X ** 3 # original dependecy 

plt.scatter(X, y)
plt.xlabel(r'$x$')
plt.ylabel(r'$y$');
X= X.reshape(-1,1)

tree = DecisionTreeRegressor(max_depth=1,max_leaf_nodes=1) #gives bug "max leaf nodes must be None or >1"
tree.fit(X, y) 

Если я опущу параметр max_leaf_nodes, я получу корневой узел и 2 листа. Нужно ли строить дерево dec с нуля? Как это сделать?

tree = DecisionTreeRegressor(max_depth=1) #works but not the same result.