add new default mappings for databricks

pantsbuild · Mar 13, 2024 · 9eee536 · 9eee536
1 parent 0c9bbce
commit 9eee536
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/python/pants/backend/python/dependency_inference/default_module_mapping.py b/src/python/pants/backend/python/dependency_inference/default_module_mapping.py
@@ -112,6 +112,11 @@ def two_groups_hyphens_two_replacements_with_suffix(
     "beautifulsoup4": ("bs4",),
     "bitvector": ("BitVector",),
     "cattrs": ("cattr", "cattrs"),
+    "databricks-sdk": ("databricks.sdk",),
+    "databricks-sql-connector": (
+        "databricks.sql",
+        "databricks.sqlalchemy",
+    ),
     "django-cors-headers": ("corsheaders",),
     "django-countries": ("django_countries",),
     "django-filter": ("django_filters",),