feat(partitions): add identity, list, range partition.py

clients/client-python/gravitino/api/expressions/partitions/identity_partition.py

@@ -17,7 +17,10 @@
 
 from abc import abstractmethod
 from typing import List, Any
+
 from .partition import Partition
+from ..literals.literal import Literal
+
 
 class IdentityPartition(Partition):
     """
@@ -41,9 +44,9 @@ def field_names(self) -> List[List[str]]:
         pass
 
     @abstractmethod
-    def values(self) -> List[Any]:
+    def values(self) -> List[Literal[Any]]:
         """
         Returns:
-            List[Any]: The values of the identity partition.
+            List[Literal[Any]]: The values of the identity partition.
         """
         pass

clients/client-python/gravitino/api/expressions/partitions/list_partition.py

@@ -18,8 +18,10 @@
 from abc import abstractmethod
 from typing import List, Any
 
+from gravitino.api.expressions.literals.literal import Literal
 from gravitino.api.expressions.partitions.partition import Partition
 
+
 class ListPartition(Partition):
     """
     A list partition represents a result of list partitioning. For example, for list partition
@@ -35,9 +37,9 @@ class ListPartition(Partition):
     """
 
     @abstractmethod
-    def lists(self) -> List[List[Any]]:
+    def lists(self) -> List[List[Literal[Any]]]:
         """
         Returns:
-            List[List[Any]]: The values of the list partition.
+            List[List[Literal[Any]]]: The values of the list partition.
         """
         pass

clients/client-python/gravitino/api/expressions/partitions/partition.py

@@ -18,6 +18,7 @@
 from abc import ABC, abstractmethod
 from typing import Dict
 
+
 class Partition(ABC):
     """
     A partition represents a result of partitioning a table. The partition can be either a

clients/client-python/gravitino/api/expressions/partitions/range_partition.py

@@ -21,6 +21,7 @@
 from gravitino.api.expressions.literals.literal import Literal
 from gravitino.api.expressions.partitions.partition import Partition
 
+
 class RangePartition(Partition):
     """
     A range partition represents a result of range partitioning. For example, for range partition