From ca3de38378e2ff3baecffa2a5bb14ed682f0f085 Mon Sep 17 00:00:00 2001
From: fxmarty <9808326+fxmarty@users.noreply.github.com>
Date: Mon, 19 Feb 2024 11:49:11 +0100
Subject: [PATCH] fix check

---
 marlin/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/marlin/__init__.py b/marlin/__init__.py
index 8b88e8f..b5b7758 100644
--- a/marlin/__init__.py
+++ b/marlin/__init__.py
@@ -80,7 +80,7 @@ def __init__(self, infeatures, outfeatures, groupsize=-1):
         super().__init__()
         if groupsize not in [-1, 128]:
             raise ValueError('Only groupsize -1 and 128 are supported.')
-        if infeatures % 128 != 0 or outfeatures != 256 == 0:
+        if infeatures % 128 != 0 or outfeatures % 256 != 0:
             raise ValueError('`infeatures` must be divisible by 128 and `outfeatures` by 256.')
         if groupsize == -1:
             groupsize = infeatures