Skip to content

Commit c1ad862

Browse files
committed
fix pylint issues
1 parent 6d0f347 commit c1ad862

2 files changed

Lines changed: 31 additions & 18 deletions

File tree

decoimpact/business/entities/rules/time_aggregation_rule.py

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,20 @@ def validate(self, logger: ILogger) -> bool:
6969
"""
7070
return self.settings.validate(self.name, logger)
7171

72+
def filter_years(
73+
self, time_dim_name: str, value_array: _xr.DataArray
74+
) -> _xr.DataArray:
75+
"""Filters the values for the specified start and end year
76+
77+
Args:
78+
value_array (DataArray): value to filter
79+
"""
80+
start = str(self._start_year) if self._start_year is not None else None
81+
end = str(self._end_year) if self._end_year is not None else None
82+
slice_obj = slice(start, end)
83+
# filtered_array = value_array.sel({time_dim_name: slice_obj})
84+
return value_array.sel({time_dim_name: slice_obj})
85+
7286
def execute(self, value_array: _xr.DataArray, logger: ILogger) -> _xr.DataArray:
7387
"""Aggregates the values for the specified start and end date
7488
@@ -99,11 +113,8 @@ def execute(self, value_array: _xr.DataArray, logger: ILogger) -> _xr.DataArray:
99113

100114
# perform aggregations in case of multi-year monthly average
101115
if TimeOperationType.MULTI_YEAR_MONTHLY_AVERAGE == settings.operation_type:
102-
start = str(self._start_year) if self._start_year is not None else None
103-
end = str(self._end_year) if self._end_year is not None else None
104-
slice_obj = slice(start, end)
105-
value_array = value_array.sel({time_dim_name: slice_obj})
106-
grouped_values = value_array.groupby(f"{time_dim_name}.month")
116+
filtered_values = self.filter_years(time_dim_name, value_array)
117+
grouped_values = filtered_values.groupby(f"{time_dim_name}.month")
107118
result = self._perform_grouping_operation(
108119
grouped_values, settings.operation_type
109120
)

decoimpact/data/entities/time_aggregation_rule_data.py

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -25,27 +25,29 @@ class TimeAggregationRuleData(TimeOperationRuleData, ITimeAggregationRuleData):
2525
def __init__(
2626
self,
2727
name: str,
28-
operation: TimeOperationType,
2928
input_variable: str,
29+
operation: TimeOperationType,
3030
start_year: Optional[int] = None,
3131
end_year: Optional[int] = None,
3232
):
3333
super().__init__(name, operation)
34+
self._name = name
35+
self._operation = operation
3436
self._input_variable = input_variable
3537
self._start_year = start_year
3638
self._end_year = end_year
3739

38-
@property
39-
def input_variable(self) -> str:
40-
"""Name of the input variable"""
41-
return self._input_variable
40+
# @property
41+
# def input_variable(self) -> str:
42+
# """Name of the input variable"""
43+
# return self._input_variable
4244

43-
@property
44-
def start_year(self) -> Optional[int]:
45-
"""Start year for aggregation"""
46-
return self._start_year
45+
# @property
46+
# def start_year(self) -> Optional[int]:
47+
# """Start year for aggregation"""
48+
# return self._start_year
4749

48-
@property
49-
def end_year(self) -> Optional[int]:
50-
"""End year for aggregation"""
51-
return self._end_year
50+
# @property
51+
# def end_year(self) -> Optional[int]:
52+
# """End year for aggregation"""
53+
# return self._end_year

0 commit comments

Comments
 (0)