style: Update ruff config (#209)

functime/_utils.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from typing import Any, Callable
 

functime/backtesting.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Any, Callable, Mapping, Optional, Tuple
 
 import numpy as np

functime/base/forecaster.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 import sys
 from dataclasses import dataclass

functime/base/metric.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from functools import wraps
 from typing import Callable, Union
 

functime/base/model.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Mapping, Protocol, Union
 

functime/base/transformer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import inspect
 import sys
 from functools import cached_property, wraps

functime/conformal.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import List, Optional
 
 import polars as pl

functime/conversion.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 import polars as pl
 

functime/cross_validation.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Mapping, Optional, Tuple, Union
 
 import numpy as np

functime/evaluation.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from functools import partial
 from typing import List, Literal, Optional
 

functime/feature_extractors.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 import math
 from typing import List, Mapping, Optional, Sequence, Union

functime/forecasting/_ar.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from typing import Any, Callable, List, Literal, Mapping, Optional, Union
 

functime/forecasting/_evaluate.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from functools import partial
 from typing import Any, Callable, List, Mapping, Optional, Tuple, Union

functime/forecasting/_reduction.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Optional
 
 import polars as pl

functime/forecasting/_regressors.py

@@ -1,6 +1,6 @@
-"""
-Fit-predict regressors with special needs.
-"""
+"""Fit-predict regressors with special needs."""
+
+from __future__ import annotations
 
 from typing import Callable, Literal, Optional, Union
 

functime/forecasting/automl.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from abc import abstractmethod
 from functools import partial
 from typing import Any, Literal, Mapping, Optional, Union

functime/forecasting/catboost.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Callable, Optional, Union
 
 import numpy as np

functime/forecasting/censored.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Callable, Optional, Union
 
 import numpy as np

functime/forecasting/elite.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from functools import partial
 from typing import Any, Literal, Mapping, Optional, Union
 

functime/forecasting/knn.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Optional
 
 import polars as pl

functime/forecasting/lance.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Literal, Optional
 
 import lance

functime/forecasting/lightgbm.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Callable, Optional, Union
 
 import numpy as np

functime/forecasting/linear.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Optional
 
 import polars as pl

functime/forecasting/naive.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Optional, Union
 
 import polars as pl

functime/forecasting/snaive.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Optional, Union
 
 import polars as pl

functime/forecasting/xgboost.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Callable, Optional, Union
 
 import numpy as np

functime/llm/api.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import List, Optional
 
 import polars as pl

functime/llm/common.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 import os
 from typing import Dict, List, Literal, Mapping

functime/llm/formatting.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Literal, Mapping, Tuple, Union
 
 import polars as pl

functime/metrics/multi_objective.py

@@ -1,5 +1,7 @@
 """Module with functions to compute, compare, and optimize multi-objective forecasts."""
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from functools import partial, reduce
 from typing import Literal, Optional

functime/metrics/point.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 import polars as pl
 

functime/offsets.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import List, Tuple, Union
 
 OFFSET_ALIASES = {"s", "m", "h", "d", "w", "mo", "y", "i"}

functime/plotting.py

@@ -1,4 +1,6 @@
-from typing import Any, Dict, Optional, Tuple, Union
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 
 import plotly.express as px
 import plotly.graph_objects as go
@@ -8,6 +10,8 @@
 from functime.base.metric import METRIC_TYPE
 from functime.metrics import smape
 
+if TYPE_CHECKING:
+    from typing import Any, Dict, Optional, Tuple, Union
 COLOR_PALETTE = {"actual": "#B7B7B7", "forecast": "#1b57f1", "backtest": "#A76EF4"}
 DEFAULT_LAST_N = 64
 

functime/preprocessing.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Any, List, Literal, Mapping, Optional, Union
 
 import cloudpickle

functime/ranges.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Optional
 
 import polars as pl

functime/seasonality/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from functime.seasonality.calendar import (
     add_calendar_effects,
     add_holiday_effects,

functime/seasonality/calendar.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import List, Literal, Optional
 
 import polars as pl

functime/seasonality/fourier.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 import polars as pl
 

functime/type_alias.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import sys
 from typing import Literal
 

pyproject.toml

@@ -117,7 +117,7 @@ select = [
   "I",  # isort
   "B",  # flake8-bugbear
   "UP", # pyupgrade
-  "I",  # isort
+  "YTT", # flake8-2020
 ]
 ignore = [
   "E501", # line too long, handled by black
@@ -127,6 +127,14 @@ ignore = [
   "F821", # Undefined name
 ]
 
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["I002"]
+"docs/**" = ["I002"]
+"**.ipynb" = ["E402"]
+
+[tool.ruff.lint.isort]
+required-imports = ["from __future__ import annotations"]
+
 [tool.ruff.lint.pyupgrade]
 # Preserve types, even if a file imports `from __future__ import annotations`.
 keep-runtime-typing = true

tests/conftest.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 import os
 from functools import partial

tests/test_benchmarks.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from functools import partial
 from typing import List, Mapping

tests/test_calendar.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from datetime import datetime
 
 import polars as pl

tests/test_changepoint_detection.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 import polars as pl
 

tests/test_cross_validation.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from contextlib import nullcontext as does_not_raise
 
 import polars as pl

tests/test_evaluation.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from functools import partial
 
 import polars as pl

tests/test_forecasting.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 
 import cloudpickle
@@ -8,15 +10,11 @@
 from functime.forecasting import (  # ann,
     auto_elastic_net,
     auto_lightgbm,
-    catboost,
     censored_model,
     elastic_net,
     elite,
-    flaml_lightgbm,
-    lightgbm,
     linear_model,
     naive,
-    xgboost,
     zero_inflated_model,
 )
 from functime.metrics import rmsse, smape, smape_original

tests/test_fourier.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from datetime import date
 
 import polars as pl

tests/test_plotting.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import polars as pl
 import pytest
 

tests/test_preprocessing.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import List, Tuple
 
 import numpy as np

tests/test_tsfresh.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import math
 
 import numpy as np