From 608cd3c0c5d34887dca94a5ac9728ed74ca2b778 Mon Sep 17 00:00:00 2001
From: Eric Johnson <ericdjohnson@google.com>
Date: Tue, 10 Jun 2025 06:55:49 -0700
Subject: [PATCH] No public description

PiperOrigin-RevId: 769613464
---
 google/colab/_shell.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/google/colab/_shell.py b/google/colab/_shell.py
index 7bf3f427..91acab10 100644
--- a/google/colab/_shell.py
+++ b/google/colab/_shell.py
@@ -89,8 +89,8 @@ def get_code_name(self, raw_code, code, number):
     """
     code_name = super().get_code_name(raw_code, code, number)
     if code_name.endswith('.py'):
-      name = pathlib.Path(code_name).with_suffix('').name
-      code_name = f'<ipython-input-{number}-{name}>'
+      path = pathlib.Path(code_name)
+      code_name = f'ipython-input-{number}-{path.name}'
     return code_name