Skip to content

Commit 2000a8d

Browse files
add parameter configuartion for aggregation representation
1 parent 250a1e1 commit 2000a8d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/main/python/systemds/scuro/representations/aggregate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def _sum_agg(data, aggregate_dim=0):
5151
def __init__(self, aggregation_function="mean", pad_modality=True, params=None):
5252
if params is not None:
5353
aggregation_function = params["aggregation_function"]
54-
pad_modality = params["pad_modality"]
54+
pad_modality = params.get("pad_modality", True)
5555

5656
if aggregation_function not in list(self._aggregation_function.keys()):
5757
raise ValueError("Invalid aggregation function")

0 commit comments

Comments
 (0)