From 01b7a570d09a8fa559cb93674c569f88f539c47b Mon Sep 17 00:00:00 2001 From: "Shah, Karan" Date: Wed, 11 Dec 2024 13:09:53 +0530 Subject: [PATCH] Review comments Signed-off-by: Shah, Karan --- openfl/callbacks/callback_list.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) 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,