Skip to content

Commit 2a4a9a1

Browse files
committed
Derived app class needs to support args and kwargs in the run function.
Signed-off-by: mmelqin <mingmelvinq@nvidia.com>
1 parent 5aabe42 commit 2a4a9a1

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

examples/apps/ai_spleen_seg_app/app.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,10 @@ def __init__(self, *args, **kwargs):
2929
self._logger = logging.getLogger("{}.{}".format(__name__, type(self).__name__))
3030
super().__init__(*args, **kwargs)
3131

32-
def run(self):
32+
def run(self, *args, **kwargs):
3333
# This method calls the base class to run. Can be omitted if simply calling through.
3434
self._logger.debug(f"Begin {self.run.__name__}")
35-
super().run()
35+
super().run(*args, **kwargs)
3636
self._logger.debug(f"End {self.run.__name__}")
3737

3838
def compose(self):

examples/apps/ai_unetr_seg_app/app.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,10 @@ def __init__(self, *args, **kwargs):
3030
self._logger = logging.getLogger("{}.{}".format(__name__, type(self).__name__))
3131
super().__init__(*args, **kwargs)
3232

33-
def run(self):
33+
def run(self, *args, **kwargs):
3434
# This method calls the base class to run. Can be omitted if simply calling through.
3535
self._logger.debug(f"Begin {self.run.__name__}")
36-
super().run()
36+
super().run(*args, **kwargs)
3737
self._logger.debug(f"End {self.run.__name__}")
3838

3939
def compose(self):

0 commit comments

Comments
 (0)