Remove size requirements for FBCNN

chaiNNer-org · Nov 18, 2023 · 605e1ab · 605e1ab
1 parent f5c2546
commit 605e1ab
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 3 deletions.
diff --git 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
diff --git a/tests/images/outputs/8x8/fbcnn_color.png b/tests/images/outputs/8x8/fbcnn_color.png
diff --git 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
@@ -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"