Add instruction to restart Python with dbutils

getyourguide · Oct 6, 2023 · a6bb34f · a6bb34f
1 parent 80bbb76
commit a6bb34f
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="engineering.data-products@getyourguide.com",
     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",