From 59a225e5d8a828bd56a9516ffcf80cac852acc2d Mon Sep 17 00:00:00 2001
From: NikoOinonen <nikooin@gmail.com>
Date: Wed, 7 Feb 2024 14:34:25 +0200
Subject: [PATCH] Removed default parameter

---
 mlspm/image/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlspm/image/models.py b/mlspm/image/models.py
index 27c0d47..b62aa38 100644
--- a/mlspm/image/models.py
+++ b/mlspm/image/models.py
@@ -52,7 +52,7 @@ class AttentionUNet(nn.Module):
 
     def __init__(
         self,
-        z_in: int = 10,
+        z_in: int,
         n_in: int = 1,
         n_out: int = 3,
         in_channels: int = 1,