Improve heuristic for Python walrus operator

* lisp/progmodes/python.el (python-base-mode): Improve heuristic for
Python walrus operator in 'electric-layout-mode'.

1 files changed, 2 insertions, 3 deletions

diff --git a/lisp/progmodes/python.el b/lisp/progmodes/python.el
index 493eab5e611..2f8d506fc60 100644
--- a/lisp/progmodes/python.el
+++ b/lisp/progmodes/python.el
@@ -7073,11 +7073,10 @@ implementations: `python-mode' and `python-ts-mode'."
               `((?: . ,(lambda ()
                          (and (zerop (car (syntax-ppss)))
                               (python-info-statement-starts-block-p)
-                              ;; Heuristic: assume walrus operator :=
-                              ;; when colon is preceded by space.
+                              ;; Heuristic for walrus operator :=
                               (save-excursion
                                 (goto-char (- (point) 2))
-                                (looking-at (rx (not space) ":")))
+                                (looking-at (rx (not space) ":" eol)))
                               'after)))))
 
   ;; Add """ ... """ pairing to electric-pair-mode.