From 40d0422e3c16ae4165bb947a2240e5f9f0e452cd Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Mon, 26 Sep 2022 14:32:48 +0200
Subject: [PATCH] add missing torch import to unet

---
 mu_map/models/unet.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mu_map/models/unet.py b/mu_map/models/unet.py
index e6e0bbc..2864d66 100644
--- a/mu_map/models/unet.py
+++ b/mu_map/models/unet.py
@@ -1,5 +1,6 @@
 from typing import Optional, List
 
+import torch
 import torch.nn as nn
 
 
-- 
GitLab