Address Fabien's comments

Author: divyashreepathihalli

keras/src/backend/common/variables.py

@@ -212,8 +212,9 @@ def __init__(
     def _deferred_initialize(self):
         if self._value is not None:
             # If NNX is enabled, it's possible the variable was already
-            # initialized by a concrete call. In this case,
-            # _deferred_initialize becomes a no-op for this variable.
+            # initialized by a concrete call.
+            if config.is_nnx_enabled():
+                return
             raise ValueError(f"Variable {self.path} is already initialized.")
 
         if in_stateless_scope():

keras/src/ops/function.py

@@ -112,11 +112,9 @@ def _setup_nnx_op_mapping(self):
         # Create a mapping from operation id to operation instance
         self._nnx_op_mapping = {}
 
-        # Store operations as direct attributes for NNX traversal
-        for i, operation in enumerate(self._operations):
-            # Store operation as direct attribute with unique name
-            attr_name = f"_op_{i}_{operation.name}"
-            setattr(self, attr_name, operation)
+        # Assign the list of operations to a single attribute for NNX traversal
+        self.nnx_operations = self._operations[:]
+        for operation in self._operations:
             # Map the operation id to this operation instance
             self._nnx_op_mapping[id(operation)] = operation