From 2f925c7728bd510a79cd918527ef2e4342c47616 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Mon, 29 Jan 2024 09:54:30 +0100
Subject: [PATCH] [test] Fix import for typehints

---
 src/mednet/utils/summary.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mednet/utils/summary.py b/src/mednet/utils/summary.py
index 2d3824c0..ffc3b417 100644
--- a/src/mednet/utils/summary.py
+++ b/src/mednet/utils/summary.py
@@ -6,7 +6,7 @@
 
 from functools import reduce
 
-import torch.nn.Module
+import torch
 
 from torch.nn.modules.module import _addindent
 
-- 
GitLab