From 8baaddc8c42d1c01738610d27660e4e0b41018d0 Mon Sep 17 00:00:00 2001 From: Oliver Neumann Date: Tue, 16 Mar 2021 16:51:01 +0100 Subject: [PATCH] Fixed bracket operator not working for steps within the pipeline. --- CHANGELOG.md | 2 ++ pywatts/core/step_information.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index c78829a4..c8e2c0be 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -13,7 +13,9 @@ ### Deprecated ### Fixed + * Fixed pipeline crashing in RMSE module because of wrong time index ([#39](https://github.com/KIT-IAI/pyWATTS/issues/39)) + * Fixed bracket operator not working for steps within the pipeline ([#42](https://github.com/KIT-IAI/pyWATTS/issues/42)) ## [0.0.1] - 2021-MM-DD diff --git a/pywatts/core/step_information.py b/pywatts/core/step_information.py index 4e454d59..43e7d39e 100644 --- a/pywatts/core/step_information.py +++ b/pywatts/core/step_information.py @@ -14,8 +14,8 @@ def __init__(self, step: BaseStep, pipeline): self.pipeline = pipeline def __getitem__(self, item: str): - from pywatts.core.pipeline_step import PipelineStep - if isinstance(self.step, PipelineStep) or len(self.step.targets) > 1: + from pywatts.core.step import Step + if isinstance(self.step, Step): self.step.last = False # TODO this should be a part of the step_factory result_step = self.step.get_result_step(item) id = self.pipeline.add(module=result_step, input_ids=[self.step.id])