Address CoPilot suggestions

Author: ezilber-akamai

linode_api4/objects/lke.py

@@ -2,6 +2,7 @@
 from typing import Any, Dict, List, Optional, Union
 from urllib import parse
 
+from linode_api4 import InstanceDiskEncryptionType
 from linode_api4.common import Price, RegionPrice
 from linode_api4.errors import UnexpectedResponseError
 from linode_api4.objects import (
@@ -422,7 +423,7 @@ def node_pool_create(
         ] = None,
         update_strategy: Optional[str] = None,
         label: str = None,
-        disk_encryption: Optional[str] = None,
+        disk_encryption: Optional[Union[str, InstanceDiskEncryptionType]] = None,
         **kwargs,
     ):
         """
@@ -445,8 +446,8 @@ def node_pool_create(
                                 NOTE: This field is specific to enterprise clusters.
         :type update_strategy: str
         :param disk_encryption: Local disk encryption setting for this LKE node pool.
-                                One of 'enabled', or 'disabled'. Defaults to 'disabled'.
-        :type disk_encryption: str
+                                One of 'enabled' or 'disabled'. Defaults to 'disabled'.
+        :type disk_encryption: str or InstanceDiskEncryptionType
         :param kwargs: Any other arguments to pass to the API.  See the API docs
                        for possible values.
 

test/integration/models/lke/test_lke.py

@@ -217,12 +217,13 @@ def test_node_pool_create_with_disk_encryption(test_linode_client, lke_cluster):
     pool = lke_cluster.node_pool_create(
         node_type,
         1,
-        disk_encryption="enabled",
+        disk_encryption=InstanceDiskEncryptionType.enabled,
     )
 
-    assert pool.disk_encryption == InstanceDiskEncryptionType.enabled
-
-    pool.delete()
+    try:
+        assert pool.disk_encryption == InstanceDiskEncryptionType.enabled
+    finally:
+        pool.delete()
 
 
 def test_cluster_dashboard_url_view(lke_cluster):