1 files changed, 8 insertions, 3 deletions
diff --git a/infra/libkookie/nixpkgs/pkgs/development/python-modules/databricks-connect/default.nix b/infra/libkookie/nixpkgs/pkgs/development/python-modules/databricks-connect/default.nix
index df3d08c3fab4..74eca940fb78 100644
--- a/infra/libkookie/nixpkgs/pkgs/development/python-modules/databricks-connect/default.nix
+++ b/infra/libkookie/nixpkgs/pkgs/development/python-modules/databricks-connect/default.nix
@@ -2,11 +2,11 @@
 
 buildPythonPackage rec {
   pname = "databricks-connect";
-  version = "7.1.1";
+  version = "7.1.10";
 
   src = fetchPypi {
     inherit pname version;
-    sha256 = "c7f508c84edc7f80a131650b892889624e4457c10f44318465dd3f7b8cf5be6d";
+    sha256 = "bae28fab1e08b709553c8efbeefc5b0ee4f5540a13a04e61fb3d348814823e16";
   };
 
   sourceRoot = ".";
@@ -16,8 +16,13 @@ buildPythonPackage rec {
   # requires network access
   doCheck = false;
 
+  prePatch = ''
+    substituteInPlace setup.py \
+      --replace "py4j==0.10.9" "py4j"
+  '';
+
   preFixup = ''
-      substituteInPlace "$out/bin/find-spark-home" \
+    substituteInPlace "$out/bin/find-spark-home" \
       --replace find_spark_home.py .find_spark_home.py-wrapped
   '';