Skip to content

Commit 7701367

Browse files
author
miranov25
committed
fix: remove relative import in fallback path (bench_groupby_regression.py)
Line 48 still had relative import in except block. Changed: from ..groupby_regression import GroupByRegressor To: from groupby_regression import GroupByRegressor After sys.path adjustment, imports must be absolute.
1 parent a7bdc58 commit 7701367

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

UTILS/dfextensions/groupby_regression/benchmarks/bench_groupby_regression.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,11 +35,17 @@
3535

3636
# --- Import the project module ---
3737
try:
38+
# Try package-relative import first (when run as module)
3839
from .. import groupby_regression as gr
3940
from ..groupby_regression import GroupByRegressor
40-
except Exception as e:
41-
print("[ERROR] Failed to import groupby_regression.py:", e, file=sys.stderr)
42-
raise
41+
except ImportError:
42+
# Fall back to adding parent to path (when run as script)
43+
script_dir = Path(__file__).parent
44+
package_dir = script_dir.parent
45+
sys.path.insert(0, str(package_dir))
46+
47+
import groupby_regression as gr
48+
from groupby_regression import GroupByRegressor
4349

4450
# --- Data Generators (Phase 1) ---
4551
def _make_groups(n_rows: int, n_groups: int, rng: np.random.Generator) -> np.ndarray:

0 commit comments

Comments
 (0)