Skip to content

Commit

Permalink
fix: deduplicate CallbackFunc class definitions (#3686)
Browse files Browse the repository at this point in the history
Signed-off-by: Kyunggeun Lee <[email protected]>
  • Loading branch information
quic-kyunggeu authored Dec 20, 2024
1 parent c6bd0bb commit 367295d
Showing 1 changed file with 2 additions and 11 deletions.
13 changes: 2 additions & 11 deletions TrainingExtensions/common/src/python/aimet_common/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,7 @@
from tqdm import tqdm
from bokeh.server.server import Server
from bokeh.application import Application
from aimet_common import defs


try:
Expand Down Expand Up @@ -97,17 +98,7 @@ class ModelApi(Enum):
onnx = 3


class CallbackFunc:
"""
Class encapsulating callback function, and it's argument(s)
"""
def __init__(self, func: Callable, func_callback_args=None):
"""
:param func: Callable Function
:param func_callback_args: Arguments passed to the callable function as-is.
"""
self.func = func
self.args = func_callback_args
CallbackFunc = defs.CallbackFunc


class SingletonType(type):
Expand Down

0 comments on commit 367295d

Please sign in to comment.