Fixed deprecated .value usage failing CI tests

Author: vfdev-5

flax/nnx/variablelib.py

@@ -1712,25 +1712,25 @@ def __contains__(self, item) -> bool:
 
   def __eq__(self, other) -> bool:
     if isinstance(other, Variable):
-      other = other.value
-    return self.value.__eq__(other)  # type: ignore
+      other = other[...]
+    return self[...].__eq__(other)  # type: ignore
 
   def __iadd__(self: V, other) -> V:
     raise NotImplementedError(
       'In-place operations are no longer supported for Variable.\n'
-      'Use `variable.value += x` instead.'
+      'Use `variable[...] += x` instead.'
     )
 
   def __isub__(self: V, other) -> V:
     raise NotImplementedError(
       'In-place operations are no longer supported for Variable.\n'
-      'Use `variable.value -= x` instead.'
+      'Use `variable[...] -= x` instead.'
     )
 
   def __imul__(self: V, other) -> V:
     raise NotImplementedError(
       'In-place operations are no longer supported for Variable.\n'
-      'Use `variable.value *= x` instead.'
+      'Use `variable[...] *= x` instead.'
     )
 
   def __imatmul__(self: V, other) -> V: