test

singlestore-labs · Dec 9, 2024 · 0db8170 · 0db8170
1 parent e12d028
commit 0db8170
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/run_shared/magic.py b/run_shared/magic.py
@@ -8,6 +8,7 @@
 from IPython.core.magic import magics_class
 from IPython.core.magic import needs_local_scope
 from IPython.core.magic import no_var_expand
+from jinja2 import Template
 
 
 @magics_class
@@ -29,7 +30,8 @@ def run_shared(self, line: str, local_ns: Any = None) -> Any:
           %run_shared shared_file.ipynb
 
         """
-        shared_file = line.strip()
+        template = Template(line.strip())
+        shared_file = template.render(local_ns)
         if not shared_file:
             raise ValueError('No shared file specified.')
         if shared_file.startswith("'") and shared_file.endswith("'"):