fix: v2 upgrade tool should ignore cell starting with '%' (#2013)

Co-authored-by: Rui Wang Napieralski <[email protected]>
Co-authored-by: Ajay Karpur <[email protected]>

Author: 3 people

src/sagemaker/cli/compatibility/v2/files.py

@@ -140,7 +140,7 @@ def _contains_shell_cmds(self, cell):
         """
         source = cell["source"]
 
-        if source[0].startswith("%%"):
+        if source[0].startswith("%"):
             return True
 
         return any(line.startswith("!") for line in source)

tests/unit/sagemaker/cli/compatibility/v2/test_file_updaters.py

@@ -114,6 +114,16 @@ def test_update(ast_transformer, pasta_parse, pasta_dump, json_dump):
           "execution_count": 3,
           "metadata": {},
           "outputs": [],
+          "source": [
+           "%%cd\\n",
+           "echo ignore this too"
+          ]
+         },
+         {
+          "cell_type": "code",
+          "execution_count": 4,
+          "metadata": {},
+          "outputs": [],
           "source": [
            "# code to be modified\\n",
            "%s"