test

singlestore-labs · Dec 9, 2024 · f2d3412 · f2d3412
1 parent e12d028
commit f2d3412
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/run_shared/magic.py b/run_shared/magic.py
@@ -1,5 +1,6 @@
 import os
 import time
+from string import Template
 from typing import Any
 
 from IPython.core.interactiveshell import InteractiveShell
@@ -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.safe_substitute(local_ns)
         if not shared_file:
             raise ValueError('No shared file specified.')
         if shared_file.startswith("'") and shared_file.endswith("'"):