Fix undesired behavior of transform.PreprocessCfg

mlfoundations · Oct 23, 2023 · dc463ae · dc463ae
1 parent 7b8dd2c
commit dc463ae
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/open_clip/transform.py b/src/open_clip/transform.py
@@ -24,15 +24,15 @@ class PreprocessCfg:
     fill_color: int = 0
 
     def __post_init__(self):
-        assert self.mode in ('RGB')
+        assert self.mode in ('RGB',)
 
     @property
     def num_channels(self):
         return 3
 
     @property
     def input_size(self):
-        return (self.num_channels(),) + to_2tuple(self.size)
+        return (self.num_channels,) + to_2tuple(self.size)
 
 _PREPROCESS_KEYS = set(asdict(PreprocessCfg()).keys())
 
@@ -403,4 +403,4 @@ def image_transform_v2(
         interpolation=cfg.interpolation,
         resize_mode=cfg.resize_mode,
         fill_color=cfg.fill_color,
-    )
+    )