diff --git a/openfl/callbacks/callback_list.py b/openfl/callbacks/callback_list.py index 47dde526898..ca88336aa3f 100644 --- a/openfl/callbacks/callback_list.py +++ b/openfl/callbacks/callback_list.py @@ -1,7 +1,5 @@ # Copyright 2020-2024 Intel Corporation # SPDX-License-Identifier: Apache-2.0 -from typing import List - from openfl.callbacks.callback import Callback from openfl.callbacks.memory_profiler import MemoryProfiler @@ -22,7 +20,7 @@ class CallbackList(Callback): def __init__( self, - callbacks: List[Callback], + callbacks: list, add_memory_profiler=False, tensor_db=None, **params,