Merge pull request #63 from getyourguide/add-extra-command-for-new-db…

…-runtime Add instruction to restart Python with dbutils
getyourguide · Oct 6, 2023 · 39db857 · 39db857
2 parents 80bbb76 + a6bb34f
commit 39db857
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,8 @@
 # Changelog db-rocket
 
+## Version 2.0.3
+- Add instruction to restart Python with dbutils (needed for newer Databricks runtimes)
+
 ## Version 2.0.2
 - fix wheel uploading to root dbfs path
 

diff --git a/rocket/rocket.py b/rocket/rocket.py
@@ -176,6 +176,7 @@ def _build_and_deploy(
 %pip install --upgrade pip
 %pip install {index_urls_options} -r {install_path}/{uploaded_dependency_file}
 %pip install --no-deps -e {install_path}
+dbutils.library.restartPython()
 
 and following in a new Python cell:
 %load_ext autoreload

diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 
 setuptools.setup(
     name="databricks-rocket",
-    version="2.0.2",
+    version="2.0.3",
     author="GetYourGuide",
     author_email="[email protected]",
     description="Keep your local python scripts installed and in sync with a databricks notebook. Shortens the feedback loop to develop projects using a hybrid enviroment",