upd

Author: MyroslavLevchyk

README.md

@@ -21,25 +21,13 @@ Here we provide some examples of how to provision it with a different options.
 # Prerequisite resources
 
 variable "databricks_account_id" {}
-variable "cloud_name" {} # cloud provider (e.g., aws, azure)
 
 # Databricks Workspace with Premium SKU
 data "azurerm_databricks_workspace" "example" {
   name                = "example-workspace"
   resource_group_name = "example-rg"
 }
 
-# Databricks Workspace ID for Azure
-data "azurerm_databricks_workspace" "example" {
-  name                = "example-workspace"
-  resource_group_name = "example-rg"
-}
-
-# Databricks Workspace ID for AWS
-data "databricks_mws_workspaces" "example" {
-  account_id = var.databricks_account_id
-}
-
 # Databricks Provider configuration
 provider "databricks" {
   alias                       = "main"
@@ -78,7 +66,7 @@ module "databricks_account_groups" {
   source  = "data-platform-hq/databricks-account-groups/databricks"
   version = "1.0.1"
   
-  workspace_id               = var.cloud_provider == "azure" ? data.azurerm_databricks_workspace.example.id : data.databricks_mws_workspaces.example.workspaces[0].workspace_id
+  workspace_id               = data.azurerm_databricks_workspace.example.id
   workspace_group_assignment = local.databricks_iam_account_groups
 
   providers = {
@@ -96,7 +84,7 @@ module "databricks_runtime_premium" {
   location = "eastus"
 
   # Cloud provider
-  cloud_name = var.cloud_name
+  cloud_name = "azure"
 
   # Example configuration for Workspace Groups
   iam_workspace_groups = {