Rename size -> size_requirements

chaiNNer-org · Nov 17, 2023 · 9803f30 · 9803f30
1 parent f9b46d4
commit 9803f30
Show file tree

Hide file tree

Showing 28 changed files with 55 additions and 53 deletions.
diff --git a/src/spandrel/__helpers/model_descriptor.py b/src/spandrel/__helpers/model_descriptor.py
@@ -46,7 +46,7 @@ def __init__(
         scale: int,
         input_channels: int,
         output_channels: int,
-        size: SizeRequirements | None = None,
+        size_requirements: SizeRequirements | None = None,
     ):
         self.model: T = model
         self.state_dict: StateDict = state_dict
@@ -59,7 +59,9 @@ def __init__(
         self.input_channels: int = input_channels
         self.output_channels: int = output_channels
 
-        self.size: SizeRequirements = size or SizeRequirements()
+        self.size_requirements: SizeRequirements = (
+            size_requirements or SizeRequirements()
+        )
 
         self.model.load_state_dict(state_dict)  # type: ignore
 
@@ -87,7 +89,7 @@ def __init__(
         supports_bfloat16: bool,
         input_channels: int,
         output_channels: int,
-        size: SizeRequirements | None = None,
+        size_requirements: SizeRequirements | None = None,
     ):
         super().__init__(
             model,
@@ -99,7 +101,7 @@ def __init__(
             scale=1,
             input_channels=input_channels,
             output_channels=output_channels,
-            size=size,
+            size_requirements=size_requirements,
         )
 
 
@@ -114,7 +116,7 @@ def __init__(
         supports_bfloat16: bool,
         input_channels: int,
         output_channels: int,
-        size: SizeRequirements | None = None,
+        size_requirements: SizeRequirements | None = None,
     ):
         super().__init__(
             model,
@@ -126,7 +128,7 @@ def __init__(
             scale=1,
             input_channels=input_channels,
             output_channels=output_channels,
-            size=size,
+            size_requirements=size_requirements,
         )
 
 

diff --git a/src/spandrel/architectures/CodeFormer/__init__.py b/src/spandrel/architectures/CodeFormer/__init__.py
@@ -58,5 +58,5 @@ def load(state_dict: StateDict) -> FaceSRModelDescriptor[CodeFormer]:
         scale=8,
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=16),
+        size_requirements=SizeRequirements(minimum=16),
     )
diff --git a/src/spandrel/architectures/DAT/__init__.py b/src/spandrel/architectures/DAT/__init__.py
@@ -157,5 +157,5 @@ def load(state_dict: StateDict) -> SRModelDescriptor[DAT]:
         scale=scale,
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=16),
+        size_requirements=SizeRequirements(minimum=16),
     )
diff --git a/src/spandrel/architectures/FBCNN/__init__.py b/src/spandrel/architectures/FBCNN/__init__.py
@@ -47,5 +47,5 @@ def load(state_dict: StateDict) -> RestorationModelDescriptor[FBCNN]:
         supports_bfloat16=True,  # TODO
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=16),  # TODO
+        size_requirements=SizeRequirements(minimum=16),  # TODO
     )
diff --git a/src/spandrel/architectures/GFPGAN/__init__.py b/src/spandrel/architectures/GFPGAN/__init__.py
@@ -45,5 +45,5 @@ def load(state_dict: StateDict) -> FaceSRModelDescriptor[GFPGANv1Clean]:
         scale=8,
         input_channels=3,
         output_channels=3,
-        size=SizeRequirements(minimum=512),
+        size_requirements=SizeRequirements(minimum=512),
     )
diff --git a/src/spandrel/architectures/HAT/__init__.py b/src/spandrel/architectures/HAT/__init__.py
@@ -181,5 +181,5 @@ def load(state_dict: StateDict) -> SRModelDescriptor[HAT]:
         scale=scale,
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=16),
+        size_requirements=SizeRequirements(minimum=16),
     )
diff --git a/src/spandrel/architectures/LaMa/__init__.py b/src/spandrel/architectures/LaMa/__init__.py
@@ -28,5 +28,5 @@ def load(state_dict: StateDict) -> InpaintModelDescriptor[LaMa]:
         supports_bfloat16=True,
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=16),
+        size_requirements=SizeRequirements(minimum=16),
     )
diff --git a/src/spandrel/architectures/MAT/__init__.py b/src/spandrel/architectures/MAT/__init__.py
@@ -26,5 +26,5 @@ def load(state_dict: StateDict) -> InpaintModelDescriptor[MAT]:
         supports_bfloat16=True,
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=512, multiple_of=512, square=True),
+        size_requirements=SizeRequirements(minimum=512, multiple_of=512, square=True),
     )
diff --git a/src/spandrel/architectures/OmniSR/__init__.py b/src/spandrel/architectures/OmniSR/__init__.py
@@ -77,5 +77,5 @@ def load(state_dict: StateDict) -> SRModelDescriptor[OmniSR]:
         scale=scale,
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=16),
+        size_requirements=SizeRequirements(minimum=16),
     )
diff --git a/src/spandrel/architectures/RestoreFormer/__init__.py b/src/spandrel/architectures/RestoreFormer/__init__.py
@@ -25,5 +25,5 @@ def load(state_dict: StateDict) -> FaceSRModelDescriptor[RestoreFormer]:
         scale=8,
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=16),
+        size_requirements=SizeRequirements(minimum=16),
     )
diff --git a/src/spandrel/architectures/SCUNet/__init__.py b/src/spandrel/architectures/SCUNet/__init__.py
@@ -33,5 +33,5 @@ def load(state_dict: StateDict) -> RestorationModelDescriptor[SCUNet]:
         supports_bfloat16=True,
         input_channels=in_nc,
         output_channels=in_nc,
-        size=SizeRequirements(minimum=16),
+        size_requirements=SizeRequirements(minimum=16),
     )
diff --git a/src/spandrel/architectures/SRFormer/__init__.py b/src/spandrel/architectures/SRFormer/__init__.py
@@ -194,5 +194,5 @@ def load(state_dict: StateDict) -> SRModelDescriptor[SRFormer]:
         scale=scale,
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=16),
+        size_requirements=SizeRequirements(minimum=16),
     )
diff --git a/src/spandrel/architectures/Swin2SR/__init__.py b/src/spandrel/architectures/Swin2SR/__init__.py
@@ -197,5 +197,5 @@ def load(state_dict: StateDict) -> SRModelDescriptor[Swin2SR]:
         scale=scale,
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=16),
+        size_requirements=SizeRequirements(minimum=16),
     )
diff --git a/src/spandrel/architectures/SwinIR/__init__.py b/src/spandrel/architectures/SwinIR/__init__.py
@@ -206,5 +206,5 @@ def load(state_dict: StateDict) -> SRModelDescriptor[SwinIR]:
         scale=scale,
         input_channels=in_nc,
         output_channels=out_nc,
-        size=SizeRequirements(minimum=16),
+        size_requirements=SizeRequirements(minimum=16),
     )
diff --git a/tests/__snapshots__/test_CodeFormer.ambr b/tests/__snapshots__/test_CodeFormer.ambr
@@ -5,7 +5,7 @@
     input_channels=3,
     output_channels=3,
     scale=8,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=False,
     tags=list([

diff --git a/tests/__snapshots__/test_Compact.ambr b/tests/__snapshots__/test_Compact.ambr
@@ -5,7 +5,7 @@
     input_channels=3,
     output_channels=3,
     scale=2,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -20,7 +20,7 @@
     input_channels=3,
     output_channels=3,
     scale=4,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([

diff --git a/tests/__snapshots__/test_ESRGAN.ambr b/tests/__snapshots__/test_ESRGAN.ambr
@@ -5,7 +5,7 @@
     input_channels=3,
     output_channels=3,
     scale=4,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -20,7 +20,7 @@
     input_channels=3,
     output_channels=3,
     scale=2,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -35,7 +35,7 @@
     input_channels=3,
     output_channels=3,
     scale=1,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -50,7 +50,7 @@
     input_channels=3,
     output_channels=3,
     scale=2,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -65,7 +65,7 @@
     input_channels=3,
     output_channels=3,
     scale=4,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -80,7 +80,7 @@
     input_channels=3,
     output_channels=3,
     scale=4,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -95,7 +95,7 @@
     input_channels=3,
     output_channels=3,
     scale=4,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -110,7 +110,7 @@
     input_channels=3,
     output_channels=3,
     scale=4,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -125,7 +125,7 @@
     input_channels=3,
     output_channels=3,
     scale=4,
-    size=SizeRequirements(minimum=None, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=None, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([

diff --git a/tests/__snapshots__/test_FBCNN.ambr b/tests/__snapshots__/test_FBCNN.ambr
@@ -5,7 +5,7 @@
     input_channels=3,
     output_channels=3,
     scale=1,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -18,7 +18,7 @@
     input_channels=1,
     output_channels=1,
     scale=1,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([

diff --git a/tests/__snapshots__/test_GFPGAN.ambr b/tests/__snapshots__/test_GFPGAN.ambr
@@ -5,7 +5,7 @@
     input_channels=3,
     output_channels=3,
     scale=8,
-    size=SizeRequirements(minimum=512, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=512, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=False,
     tags=list([
@@ -18,7 +18,7 @@
     input_channels=3,
     output_channels=3,
     scale=8,
-    size=SizeRequirements(minimum=512, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=512, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=False,
     tags=list([
@@ -31,7 +31,7 @@
     input_channels=3,
     output_channels=3,
     scale=8,
-    size=SizeRequirements(minimum=512, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=512, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=False,
     tags=list([

diff --git a/tests/__snapshots__/test_HAT.ambr b/tests/__snapshots__/test_HAT.ambr
@@ -5,7 +5,7 @@
     input_channels=3,
     output_channels=3,
     scale=4,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=False,
     tags=list([

diff --git a/tests/__snapshots__/test_LaMa.ambr b/tests/__snapshots__/test_LaMa.ambr
@@ -5,7 +5,7 @@
     input_channels=4,
     output_channels=3,
     scale=1,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=False,
     tags=list([

diff --git a/tests/__snapshots__/test_MAT.ambr b/tests/__snapshots__/test_MAT.ambr
@@ -5,7 +5,7 @@
     input_channels=3,
     output_channels=3,
     scale=1,
-    size=SizeRequirements(minimum=512, multiple_of=512, square=True),
+    size_requirements=SizeRequirements(minimum=512, multiple_of=512, square=True),
     supports_bfloat16=True,
     supports_half=False,
     tags=list([

diff --git a/tests/__snapshots__/test_OmniSR.ambr b/tests/__snapshots__/test_OmniSR.ambr
@@ -5,7 +5,7 @@
     input_channels=3,
     output_channels=3,
     scale=2,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -21,7 +21,7 @@
     input_channels=3,
     output_channels=3,
     scale=4,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([

diff --git a/tests/__snapshots__/test_RestoreFormer.ambr b/tests/__snapshots__/test_RestoreFormer.ambr
@@ -5,7 +5,7 @@
     input_channels=3,
     output_channels=3,
     scale=8,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=False,
     tags=list([

diff --git a/tests/__snapshots__/test_SCUNet.ambr b/tests/__snapshots__/test_SCUNet.ambr
@@ -5,7 +5,7 @@
     input_channels=3,
     output_channels=3,
     scale=1,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -18,7 +18,7 @@
     input_channels=3,
     output_channels=3,
     scale=1,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -31,7 +31,7 @@
     input_channels=3,
     output_channels=3,
     scale=1,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([
@@ -44,7 +44,7 @@
     input_channels=1,
     output_channels=1,
     scale=1,
-    size=SizeRequirements(minimum=16, multiple_of=None, square=False),
+    size_requirements=SizeRequirements(minimum=16, multiple_of=None, square=False),
     supports_bfloat16=True,
     supports_half=True,
     tags=list([