From 013a8bfef3a63bb110ffcb61bbf227a33777d180 Mon Sep 17 00:00:00 2001 From: YunLiu <55491388+KumoLiu@users.noreply.github.com> Date: Thu, 29 Feb 2024 10:26:59 +0800 Subject: [PATCH] remove self._probs in initialize Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com> --- bundle/python_bundle_workflow/scripts/inference.py | 2 +- bundle/python_bundle_workflow/scripts/train.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/bundle/python_bundle_workflow/scripts/inference.py b/bundle/python_bundle_workflow/scripts/inference.py index fab321b009..19f9aa21ee 100644 --- a/bundle/python_bundle_workflow/scripts/inference.py +++ b/bundle/python_bundle_workflow/scripts/inference.py @@ -66,7 +66,7 @@ def __init__(self, dataset_dir: str = "."): self.dataset_dir = dataset_dir def initialize(self): - self.props = {} + pass def run(self): self.evaluator.run() diff --git a/bundle/python_bundle_workflow/scripts/train.py b/bundle/python_bundle_workflow/scripts/train.py index 56fc36056a..bc8a47afa6 100644 --- a/bundle/python_bundle_workflow/scripts/train.py +++ b/bundle/python_bundle_workflow/scripts/train.py @@ -81,7 +81,6 @@ def __init__(self, dataset_dir: str = "."): def initialize(self): set_determinism(0) - self.props = {} def run(self): self.trainer.run()