diff --git a/pvnet_summation/data/datamodule.py b/pvnet_summation/data/datamodule.py index 0f5f0e5..c8a92a3 100644 --- a/pvnet_summation/data/datamodule.py +++ b/pvnet_summation/data/datamodule.py @@ -105,7 +105,7 @@ def __init__(self, **datapipes): def __iter__(self): for outputs in self.source_datapipes: - yield {key: value for key, value in zip(self.keys, outputs)} # noqa: B905 + yield {key: value for key, value in zip(self.keys, outputs)} # noqa: B905 def get_capacity(batch): diff --git a/pvnet_summation/models/model.py b/pvnet_summation/models/model.py index 5256be8..42b426f 100644 --- a/pvnet_summation/models/model.py +++ b/pvnet_summation/models/model.py @@ -13,7 +13,6 @@ from pvnet_summation.models.base_model import BaseModel - _default_optimizer = pvnet.optimizers.Adam() @@ -54,7 +53,7 @@ def __init__( self.scale_pvnet_outputs = scale_pvnet_outputs self.predict_difference_from_sum = predict_difference_from_sum - + if output_network_kwargs is None: output_network_kwargs = dict()