From 0bb02aa78b7123874547092d3b1bd53ac6d5be5e Mon Sep 17 00:00:00 2001
From: Katlyn Edwards <katlyn@google.com>
Date: Mon, 20 May 2024 19:04:09 -0700
Subject: [PATCH] No public description

PiperOrigin-RevId: 635638314
---
 google/colab/_shell_customizations.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/google/colab/_shell_customizations.py b/google/colab/_shell_customizations.py
index a5c0d46f..06eed5f6 100644
--- a/google/colab/_shell_customizations.py
+++ b/google/colab/_shell_customizations.py
@@ -152,7 +152,7 @@ def import_message(error):
                   'action_text': 'Install {}'.format(module_name),
                   # Snippets for installing a custom library always end with
                   # an import of the library itself.
-                  'snippet_filter': 'import {}'.format(module_name),
+                  'snippet_filter': module_name,
               },
           ],
       }