cleaned

sgoldenlab · Dec 12, 2024 · 11d64a8 · 11d64a8
1 parent 25fecdd
commit 11d64a8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/simba/model/regression/model.py b/simba/model/regression/model.py
@@ -1,10 +1,11 @@
-from typing import Dict, List, Optional, Tuple
 from itertools import product
+from typing import Dict, List, Optional, Tuple
 
 import numpy as np
 import pandas as pd
 import xgboost as xgb
 from sklearn.model_selection import StratifiedKFold
+
 from simba.model.regression.metrics import (mean_absolute_error,
                                             mean_absolute_percentage_error,
                                             mean_squared_error, r2_score,
@@ -15,6 +16,7 @@
 from simba.utils.enums import Formats
 from simba.utils.errors import DataHeaderError
 
+
 def fit_xgb(x: pd.DataFrame,
             y: np.ndarray,
             xgb_reg: xgb.XGBRegressor) -> xgb.XGBRegressor: