Skip to content

Commit 8ca5e2d

Browse files
author
miranov25
committed
fix: Resolve 4 pylint errors
- Fix function-redefined in test_groupby_regression.py - Add LinAlgError import in groupby_regression_optimized.py - Fix import paths in sliding window test files
1 parent 0c098be commit 8ca5e2d

File tree

4 files changed

+5
-3
lines changed

4 files changed

+5
-3
lines changed

UTILS/dfextensions/groupby_regression/groupby_regression_optimized.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -413,6 +413,8 @@ def make_parallel_fit_v2(
413413
# ======================================================================
414414

415415
import numpy as np
416+
from numpy.linalg import LinAlgError
417+
from numpy.linalg import LinAlgError
416418
import pandas as pd
417419
import time
418420

UTILS/dfextensions/groupby_regression/tests/test_groupby_regression.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,7 @@ def test_sigma_cut_impact():
290290

291291

292292

293-
def test_make_parallel_fit_robust(sample_data):
293+
def test_make_parallel_fit_robust_v2(sample_data):
294294
df = sample_data.copy()
295295
df_out, dfGB = GroupByRegressor.make_parallel_fit(
296296
df,

UTILS/dfextensions/groupby_regression/tests/test_groupby_regression_sliding_window.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -779,7 +779,7 @@ def test_window_size_zero_parity_with_v4_relaxed():
779779
Establishes continuity with v4 when sliding window is disabled.
780780
"""
781781
try:
782-
from ..groupby_regression_optimized import make_parallel_fit as make_parallel_fit_v4
782+
from ..groupby_regression import make_parallel_fit as make_parallel_fit_v4
783783
except Exception:
784784
pytest.skip("v4 not available for comparison")
785785

UTILS/dfextensions/groupby_regression/tests/test_groupby_regression_sliding_window_verbose.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -849,7 +849,7 @@ def test_window_size_zero_parity_with_v4_relaxed():
849849
Establishes continuity with v4 when sliding window is disabled.
850850
"""
851851
try:
852-
from ..groupby_regression_optimized import make_parallel_fit as make_parallel_fit_v4
852+
from ..groupby_regression import make_parallel_fit as make_parallel_fit_v4
853853
except Exception:
854854
pytest.skip("v4 not available for comparison")
855855

0 commit comments

Comments
 (0)