Small fix

Signed-off-by: Chaurasiya, Payal <[email protected]>
payalcha · Dec 11, 2024 · f5cd467 · f5cd467
1 parent 8d20e42
commit f5cd467
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openfl/utilities/logs.py b/openfl/utilities/logs.py
@@ -11,7 +11,7 @@
 import psutil
 from rich.console import Console
 from rich.logging import RichHandler
-from torch.utils.tensorboard import SummaryWriter
+from tensorboardX import SummaryWriter
 
 
 def write_metric(node_name, task_name, metric_name, metric, round_number):