From 605e1ab933672deabe3f874cada623c1d669120f Mon Sep 17 00:00:00 2001
From: Michael Schmidt <mitchi5000.ms@googlemail.com>
Date: Sat, 18 Nov 2023 11:30:18 +0100
Subject: [PATCH] Remove size requirements for FBCNN

---
 src/spandrel/architectures/FBCNN/__init__.py |   2 --
 tests/images/inputs/8x8.png                  | Bin 0 -> 183 bytes
 tests/images/outputs/8x8/fbcnn_color.png     | Bin 0 -> 268 bytes
 tests/test_FBCNN.py                          |   2 +-
 tests/util.py                                |   1 +
 5 files changed, 2 insertions(+), 3 deletions(-)
 create mode 100644 tests/images/inputs/8x8.png
 create mode 100644 tests/images/outputs/8x8/fbcnn_color.png

diff --git a/src/spandrel/architectures/FBCNN/__init__.py b/src/spandrel/architectures/FBCNN/__init__.py
index 044c8e97..1ed53276 100644
--- a/src/spandrel/architectures/FBCNN/__init__.py
+++ b/src/spandrel/architectures/FBCNN/__init__.py
@@ -1,6 +1,5 @@
 from ...__helpers.model_descriptor import (
     RestorationModelDescriptor,
-    SizeRequirements,
     StateDict,
 )
 from .arch.FBCNN import FBCNN
@@ -47,5 +46,4 @@ def load(state_dict: StateDict) -> RestorationModelDescriptor[FBCNN]:
         supports_bfloat16=True,  # TODO
         input_channels=in_nc,
         output_channels=out_nc,
-        size_requirements=SizeRequirements(minimum=16),  # TODO
     )
diff --git a/tests/images/inputs/8x8.png b/tests/images/inputs/8x8.png
new file mode 100644
index 0000000000000000000000000000000000000000..d45e67e886070820eae94d4e84337c50ca33cc73
GIT binary patch
literal 183
zcmeAS@N?(olHy`uVBq!ia0vp^93afX3?$7I7w-U4jKx9jP7LeL$-D$|SkfJR9T^xl
z_H+M9WCij?1AIbUHMf2HkTmIyLzSYUqLgyoSBCJlBH`;bn}mQ0I14-?iy0XB4ude`
z@%$AjKtXL!7sn8Z%gGH4j*e_>ZjO$OY-)*#ifpcef{JWQ9S$<EnW-LRVcQn4je$Ys
W5;Lc;=D`O*Z492SelF{r5}E*DH!a%$

literal 0
HcmV?d00001

diff --git a/tests/images/outputs/8x8/fbcnn_color.png b/tests/images/outputs/8x8/fbcnn_color.png
new file mode 100644
index 0000000000000000000000000000000000000000..cbe62b4c0c25d43a966342af0bb30559ba7e115d
GIT binary patch
literal 268
zcmV+n0rUQeP)<h;3K|Lk000e1NJLTq000O8000OG0ssI2OKmCK0002fNkl<Z2ps{)
z05|^uM5gHh0|FcP@Dcd*0RIK!2o&c93<Ul80ZXg$1o-$Z@Zl)%;Qs&y*%Knq3?BIi
z4FPSO>=o(cMC;cj?%eYW2Jrm{-5e{}6C44ve8nc=*dqAtw5y*qT2a8Ju|r&A$N~ld
z%1eGV3=!q}_p7+MO<hjAp|3zmOW+e00l!d)DCy$)@%Z*KG71q4`0)13+1w9FPXV%%
z&gztpT?P#L5g7t2BIEY@q|eYNU}gb_gtzIau2AOZA_xZ&Lq@L6&C?AKAWu>b)?nk?
SlEa(;0000<MNUMnLSTYEGI5mv

literal 0
HcmV?d00001

diff --git a/tests/test_FBCNN.py b/tests/test_FBCNN.py
index d01b7807..8e665706 100644
--- a/tests/test_FBCNN.py
+++ b/tests/test_FBCNN.py
@@ -11,7 +11,7 @@ def test_FBCNN_color(snapshot):
     model = ModelLoader().load_from_file(file.path)
     assert model == snapshot(exclude=disallowed_props)
     assert isinstance(model.model, FBCNN)
-    assert_image_inference(file, model, [TestImage.JPEG_15])
+    assert_image_inference(file, model, [TestImage.JPEG_15, TestImage.SR_8])
 
 
 def test_FBCNN_gray(snapshot):
diff --git a/tests/util.py b/tests/util.py
index 2d5d3256..d8b8661c 100644
--- a/tests/util.py
+++ b/tests/util.py
@@ -111,6 +111,7 @@ def get_h_w_c(image: np.ndarray) -> tuple[int, int, int]:
 
 
 class TestImage(Enum):
+    SR_8 = "8x8.png"
     SR_16 = "16x16.png"
     SR_32 = "32x32.png"
     SR_64 = "64x64.png"