Fix test

Signed-off-by: Merel Theisen <[email protected]>

Author: merelcht

kedro-datasets/kedro_datasets_experimental/databricks/external_table_dataset.py

@@ -75,46 +75,46 @@ class ExternalTableDataset(BaseTableDataset):
 
     ### Example usage for the [Python API](https://docs.kedro.org/en/stable/catalog-data/advanced_data_catalog_usage/):
 
-```python
-from kedro_datasets.databricks import ExternalTableDataset
-from pyspark.sql import SparkSession
-from pyspark.sql.types import IntegerType, Row, StringType, StructField, StructType
-import importlib.metadata
-
-DELTA_VERSION = importlib.metadata.version("delta-spark")
-major_version = int(DELTA_VERSION.split(".")[0])
-delta_package = (
-    f"io.delta:delta-spark_2.13:{DELTA_VERSION}"
-    if major_version >= 4
-    else f"io.delta:delta-core_2.12:{DELTA_VERSION}"
-)
-
-schema = StructType(
-    [StructField("name", StringType(), True), StructField("age", IntegerType(), True)]
-)
-data = [("Alex", 31), ("Bob", 12), ("Clarke", 65), ("Dave", 29)]
-
-spark_df = (
-    SparkSession.builder.config("spark.jars.packages", delta_package)
-    .config("spark.sql.extensions", "io.delta.sql.DeltaSparkSessionExtension")
-    .config(
-        "spark.sql.catalog.spark_catalog",
-        "org.apache.spark.sql.delta.catalog.DeltaCatalog",
+    ```python
+    from kedro_datasets.databricks import ExternalTableDataset
+    from pyspark.sql import SparkSession
+    from pyspark.sql.types import IntegerType, Row, StringType, StructField, StructType
+    import importlib.metadata
+
+    DELTA_VERSION = importlib.metadata.version("delta-spark")
+    major_version = int(DELTA_VERSION.split(".")[0])
+    delta_package = (
+        f"io.delta:delta-spark_2.13:{DELTA_VERSION}"
+        if major_version >= 4
+        else f"io.delta:delta-core_2.12:{DELTA_VERSION}"
     )
-    .getOrCreate()
-    .createDataFrame(data, schema)
-)
-
-dataset = ExternalTableDataset(
-    table="names_and_ages",
-    write_mode="overwrite",
-    location="abfss://[email protected]/depts/cust",
-)
-
-dataset.save(spark_df)
-reloaded = dataset.load()
-assert Row(name="Bob", age=12) in reloaded.take(4)
-```
+
+    schema = StructType(
+        [StructField("name", StringType(), True), StructField("age", IntegerType(), True)]
+    )
+    data = [("Alex", 31), ("Bob", 12), ("Clarke", 65), ("Dave", 29)]
+
+    spark_df = (
+        SparkSession.builder.config("spark.jars.packages", delta_package)
+        .config("spark.sql.extensions", "io.delta.sql.DeltaSparkSessionExtension")
+        .config(
+            "spark.sql.catalog.spark_catalog",
+            "org.apache.spark.sql.delta.catalog.DeltaCatalog",
+        )
+        .getOrCreate()
+        .createDataFrame(data, schema)
+    )
+
+    dataset = ExternalTableDataset(
+        table="names_and_ages",
+        write_mode="overwrite",
+        location="abfss://[email protected]/depts/cust",
+    )
+
+    dataset.save(spark_df)
+    reloaded = dataset.load()
+    assert Row(name="Bob", age=12) in reloaded.take(4)
+    ```
 
     """