From 00aa6f9b973805c0f7873ac1036f044ab2702ce1 Mon Sep 17 00:00:00 2001
From: Dalton Worsnup <60011797+DaltonWorsnup@users.noreply.github.com>
Date: Fri, 24 Oct 2025 09:38:52 -0700
Subject: [PATCH] Correct MemoryFormat.set() Parameter Type Annotation

The type annotation on the parameter for MemoryFormat.set() was previously set to EngineLiteral. This change corrects the type annotation to MemoryFormatLiteral
---
 awswrangler/_distributed.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/awswrangler/_distributed.py b/awswrangler/_distributed.py
index aa0735d38..dffb85816 100644
--- a/awswrangler/_distributed.py
+++ b/awswrangler/_distributed.py
@@ -177,7 +177,7 @@ def get(cls) -> MemoryFormatEnum:
             return cls._enum if cls._enum else cls.get_installed()
 
     @classmethod
-    def set(cls, name: EngineLiteral) -> None:
+    def set(cls, name: MemoryFormatLiteral) -> None:
         """Set the memory format."""
         with cls._lock:
             cls._enum = MemoryFormatEnum[name.upper()]