Remove Linked List Elements Python Solution

lcpy gen -n 203   # by problem number
lcpy gen -s remove_linked_list_elements   # by problem name

Input: head = [1,2,6,3,4,5,6], val = 6
Output: [1,2,3,4,5]

Input: head = [], val = 1
Output: []

Input: head = [7,7,7,7], val = 7
Output: []

from leetcode_py import ListNode


class Solution:
    # Time: O(n)
    # Space: O(1)
    def remove_linked_list_elements(
        self, head: ListNode[int] | None, val: int
    ) -> ListNode[int] | None:
        dummy = ListNode[int](0)
        dummy.next = head
        current = dummy
        while current.next is not None:
            if current.next.val == val:
                current.next = current.next.next
            else:
                current = current.next
        return dummy.next