Address CoPilot suggestions

Author: ezilber-akamai

linode_api4/objects/lke.py

@@ -15,6 +15,7 @@
     Type,
 )
 from linode_api4.objects.base import _flatten_request_body_recursive
+from linode_api4.objects.linode import InstanceDiskEncryptionType
 from linode_api4.util import drop_null_keys
 
 
@@ -422,7 +423,9 @@ 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 +448,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

@@ -210,18 +210,20 @@ def _to_comparable(p: LKENodePool) -> Dict[str, Any]:
         InstanceDiskEncryptionType.disabled,
     )
 
+
 def test_node_pool_create_with_disk_encryption(test_linode_client, lke_cluster):
     node_type = test_linode_client.linode.types()[1]
 
     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):