Skip to content

Commit

Permalink
remove self._probs in initialize
Browse files Browse the repository at this point in the history
Signed-off-by: YunLiu <[email protected]>
  • Loading branch information
KumoLiu committed Feb 29, 2024
1 parent 3be3f85 commit 013a8bf
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 2 deletions.
2 changes: 1 addition & 1 deletion bundle/python_bundle_workflow/scripts/inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down
1 change: 0 additions & 1 deletion bundle/python_bundle_workflow/scripts/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,6 @@ def __init__(self, dataset_dir: str = "."):

def initialize(self):
set_determinism(0)
self.props = {}

def run(self):
self.trainer.run()
Expand Down

0 comments on commit 013a8bf

Please sign in to comment.