Skip to content

Commit 0fded6b

Browse files
Fix indentation errors
1 parent 6feddf5 commit 0fded6b

File tree

1 file changed

+23
-22
lines changed

1 file changed

+23
-22
lines changed

lazypredict/Supervised.py

Lines changed: 23 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ class LazyClassifier:
163163
prediction : bool, optional (default=False)
164164
When set to True, the predictions of all the models models are returned as dataframe.
165165
classifiers : list, optional (default="all")
166-
When function is provided, trains the chosen classifier(s).
166+
When function is provided, trains the chosen classifier(s).
167167
168168
Examples
169169
--------
@@ -285,17 +285,18 @@ def fit(self, X_train, X_test, y_train, y_test):
285285
)
286286

287287
if self.classifiers == "all":
288-
self.classifiers = CLASSIFIERS
288+
self.classifiers = CLASSIFIERS
289289
else:
290-
try:
291-
temp_list=[]
292-
for classifier in self.classifiers:
293-
full_name = (classifier.__class__.__name__, classifier)
294-
temp_list.append(full_name)
295-
self.classifiers = temp_list
296-
except Exception as exception:
297-
print(exception)
298-
print("Invalid Classifier(s)")
290+
try:
291+
temp_list = []
292+
for classifier in self.classifiers:
293+
full_name = (classifier.__class__.__name__, classifier)
294+
temp_list.append(full_name)
295+
self.classifiers = temp_list
296+
except Exception as exception:
297+
print(exception)
298+
print("Invalid Classifier(s)")
299+
299300
for name, model in tqdm(self.classifiers):
300301
start = time.time()
301302
try:
@@ -446,7 +447,7 @@ class LazyRegressor:
446447
prediction : bool, optional (default=False)
447448
When set to True, the predictions of all the models models are returned as dataframe.
448449
regressors : list, optional (default="all")
449-
When function is provided, trains the chosen regressor(s).
450+
When function is provided, trains the chosen regressor(s).
450451
451452
Examples
452453
--------
@@ -586,17 +587,17 @@ def fit(self, X_train, X_test, y_train, y_test):
586587
)
587588

588589
if self.regressors == "all":
589-
self.regressors = REGRESSORS
590+
self.regressors = REGRESSORS
590591
else:
591-
try:
592-
temp_list = []
593-
for regressor in self.regressors:
594-
full_name = (regressor.__class__.__name__, regressor)
595-
temp_list.append(full_name)
596-
self.regressors = temp_list
597-
except Exception as exception:
598-
print(exception)
599-
print("Invalid Regressor(s)")
592+
try:
593+
temp_list = []
594+
for regressor in self.regressors:
595+
full_name = (regressor.__class__.__name__, regressor)
596+
temp_list.append(full_name)
597+
self.regressors = temp_list
598+
except Exception as exception:
599+
print(exception)
600+
print("Invalid Regressor(s)")
600601

601602
for name, model in tqdm(self.regressors):
602603
start = time.time()

0 commit comments

Comments
 (0)