fix bug

src/codegen/sdk/core/symbol_groups/collection.py

@@ -19,7 +19,7 @@
 Parent = TypeVar("Parent")
 
 
 class Collection(SymbolGroup[Child, Parent], MutableSequence[Child], Generic[Child, Parent]):
     """Ordered collection of nodes
     Attributes:
         _bracket_size: Number of characters wrapping the collection
@@ -47,7 +47,7 @@
         if children is not None:
             self._init_children(children)
 
     def _init_children(self, symbols: list[Child]):
         """Call this after setting self._symbols."""
         if self.ts_node.start_point[0] != self.ts_node.end_point[0] and symbols:
             # This is a multiline collection.
@@ -70,7 +70,7 @@
         if isinstance(key, slice):
             assert isinstance(value, Iterable)
             for idx, item in zip(range(key.start, key.stop, key.step), value):
                 self[idx] = item
         else:
             assert not isinstance(value, Iterable)
             if isinstance(value, Editable):
@@ -94,7 +94,7 @@
         return self._elements + self._inserts_till()
 
     @writer
     def remove(self, value: Child | None = None, *args, **kwargs) -> None:
         """Removes an element from a Collection.
 
         Deletes the specified element from the Collection by calling its remove method. If no value is specified,
@@ -112,7 +112,8 @@
         # For example, let's remove all occurrences of the value instead of just the first one
         if value is None:
             super().remove(*args, **kwargs)
-        value.remove(*args, **kwargs)
+        else:
+            value.remove(*args, **kwargs)
 
     def _inserts_till(self, max_idx: int | None = None) -> int:
         """Find the number of pending inserts until max_idx."""
@@ -240,7 +241,7 @@
         Returns:
             None
         """
         return Editable.edit(self, *args, **kwargs)  # HACK: keep start/end brackets
 
     @property
     @reader
@@ -276,4 +277,4 @@
         self._reversed.clear()
 
     def _smart_remove(self, child, *args, **kwargs) -> bool:
         return self.parent._smart_remove(self, child, *args, **kwargs)