use strip to determine cell magic

kiyoon · Jul 2, 2023 · 2bdfe42 · 2bdfe42
1 parent 3cc5a85
commit 2bdfe42
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lua/jupynium/cells.lua b/lua/jupynium/cells.lua
@@ -16,7 +16,7 @@ function M.line_type(line)
     return "cell separator: markdown (string)"
   elseif utils.string_begins_with(line, "# %% [md]") or utils.string_begins_with(line, "# %% [markdown]") then
     return "cell separator: markdown (jupytext)"
-  elseif utils.string_ends_with(line, "# %%") then
+  elseif vim.fn.trim(line)=="# %%" then
     return "cell separator: code"
   elseif utils.string_begins_with(line, '%%"""') or utils.string_begins_with(line, "%%'''") then
     return "cell separator: code (string)"

diff --git a/src/jupynium/buffer.py b/src/jupynium/buffer.py
@@ -91,7 +91,7 @@ def full_analyse_buf(self, header_cell_type="header"):
                 num_rows_this_cell = 1
                 cell_types.append("markdown (jupytext)")
             elif (
-                line[:5] == "# %%"
+                line.strip() == "# %%"
                 or line.startswith('%%"""')
                 or line.startswith("%%'''")
             ):