Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 8 additions & 5 deletions dte_adj/__init__.py
Original file line number Diff line number Diff line change
@@ -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",
Expand Down
4 changes: 2 additions & 2 deletions dte_adj/local.py
Original file line number Diff line number Diff line change
@@ -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):
Expand Down
2 changes: 1 addition & 1 deletion dte_adj/simple.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
import numpy as np
from .stratified import (
from dte_adj.stratified import (
SimpleStratifiedDistributionEstimator,
AdjustedStratifiedDistributionEstimator,
)
Expand Down
2 changes: 1 addition & 1 deletion dte_adj/stratified.py
Original file line number Diff line number Diff line change
@@ -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):
Expand Down
8 changes: 8 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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"
4 changes: 2 additions & 2 deletions uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading