From 534206b7ddb8d611083a9d2bc797cf34243472ad Mon Sep 17 00:00:00 2001 From: TomeHirata Date: Wed, 24 Sep 2025 20:08:24 +0900 Subject: [PATCH] Use absolute imports --- dte_adj/__init__.py | 13 ++++++++----- dte_adj/local.py | 4 ++-- dte_adj/simple.py | 2 +- dte_adj/stratified.py | 2 +- pyproject.toml | 8 ++++++++ uv.lock | 4 ++-- 6 files changed, 22 insertions(+), 11 deletions(-) diff --git a/dte_adj/__init__.py b/dte_adj/__init__.py index fff0ba7..be804ec 100644 --- a/dte_adj/__init__.py +++ b/dte_adj/__init__.py @@ -1,14 +1,17 @@ # Import estimator classes from separate modules -from .base import DistributionEstimatorBase -from .simple import SimpleDistributionEstimator, AdjustedDistributionEstimator -from .stratified import ( +from dte_adj.base import DistributionEstimatorBase +from dte_adj.simple import SimpleDistributionEstimator, AdjustedDistributionEstimator +from dte_adj.stratified import ( SimpleStratifiedDistributionEstimator, AdjustedStratifiedDistributionEstimator, ) -from .local import SimpleLocalDistributionEstimator, AdjustedLocalDistributionEstimator +from dte_adj.local import ( + SimpleLocalDistributionEstimator, + AdjustedLocalDistributionEstimator, +) # Import utility functions -from .util import compute_confidence_intervals, compute_ldte, compute_lpte +from dte_adj.util import compute_confidence_intervals, compute_ldte, compute_lpte __all__ = [ "DistributionEstimatorBase", diff --git a/dte_adj/local.py b/dte_adj/local.py index 596d680..d3622ee 100644 --- a/dte_adj/local.py +++ b/dte_adj/local.py @@ -1,10 +1,10 @@ import numpy as np from typing import Tuple -from .stratified import ( +from dte_adj.stratified import ( SimpleStratifiedDistributionEstimator, AdjustedStratifiedDistributionEstimator, ) -from .util import compute_ldte, compute_lpte +from dte_adj.util import compute_ldte, compute_lpte class SimpleLocalDistributionEstimator(SimpleStratifiedDistributionEstimator): diff --git a/dte_adj/simple.py b/dte_adj/simple.py index 160bd00..3b86a3a 100644 --- a/dte_adj/simple.py +++ b/dte_adj/simple.py @@ -1,5 +1,5 @@ import numpy as np -from .stratified import ( +from dte_adj.stratified import ( SimpleStratifiedDistributionEstimator, AdjustedStratifiedDistributionEstimator, ) diff --git a/dte_adj/stratified.py b/dte_adj/stratified.py index e067eb3..e7e3d30 100644 --- a/dte_adj/stratified.py +++ b/dte_adj/stratified.py @@ -1,7 +1,7 @@ import numpy as np from typing import Tuple, Any from copy import deepcopy -from .base import DistributionEstimatorBase +from dte_adj.base import DistributionEstimatorBase class SimpleStratifiedDistributionEstimator(DistributionEstimatorBase): diff --git a/pyproject.toml b/pyproject.toml index 841c483..5488186 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -49,3 +49,11 @@ dev-dependencies = [ "scikit-learn>=1.5,<1.8", "pre-commit>=4.0.1,<4.4.0" ] + +[tool.ruff.lint] +# TID252: Ban relative imports - enforce absolute imports for internal modules +select = ["TID252"] + +[tool.ruff.lint.flake8-tidy-imports] +# Ban all relative imports +ban-relative-imports = "all" diff --git a/uv.lock b/uv.lock index 05e5f37..f4096cf 100644 --- a/uv.lock +++ b/uv.lock @@ -346,7 +346,7 @@ requires-dist = [ { name = "matplotlib", specifier = ">=3.7" }, { name = "numpy", specifier = ">=1.26" }, { name = "pre-commit", marker = "extra == 'dev'", specifier = ">=4.0.1,<4.4.0" }, - { name = "ruff", marker = "extra == 'dev'", specifier = "~=0.12.2" }, + { name = "ruff", marker = "extra == 'dev'", specifier = ">=0.12.2,<0.14.0" }, { name = "scikit-learn", marker = "extra == 'dev'", specifier = ">=1.5,<1.8" }, { name = "scipy", specifier = ">=1.13.1" }, { name = "sphinx", marker = "extra == 'dev'", specifier = ">=7.3.7,<8.2.0" }, @@ -357,7 +357,7 @@ provides-extras = ["dev"] dev = [ { name = "build", specifier = ">=1.2.2,<1.4.0" }, { name = "pre-commit", specifier = ">=4.0.1,<4.4.0" }, - { name = "ruff", specifier = "~=0.12.2" }, + { name = "ruff", specifier = ">=0.12.2,<0.14.0" }, { name = "scikit-learn", specifier = ">=1.5,<1.8" }, { name = "sphinx", specifier = ">=7.3.7,<8.2.0" }, ]