diff --git a/requirements.txt b/requirements.txt
index 3a2e7ce..e0db8e8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,5 @@ databricks-api
 requests
 fire
 junit_xml
+py4j