feat: add __slots__ to Series, DataFrame, Categorical and Enum

pola-rs · Feb 17, 2024 · 4ed1b8b · 4ed1b8b
1 parent d04a35f
commit 4ed1b8b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/py-polars/polars/dataframe/frame.py b/py-polars/polars/dataframe/frame.py
@@ -348,6 +348,8 @@ class DataFrame:
     False
     """
 
+    __slots__ = ("_df",)
+    _df: PyDataFrame
     _accessors: ClassVar[set[str]] = {"plot"}
 
     def __init__(

diff --git a/py-polars/polars/datatypes/classes.py b/py-polars/polars/datatypes/classes.py
@@ -237,6 +237,7 @@ def _custom_reconstruct(
 class DataTypeGroup(frozenset):  # type: ignore[type-arg]
     """Group of data types."""
 
+    __slots__ = ("_match_base_type",)
     _match_base_type: bool
 
     def __new__(
@@ -379,7 +380,6 @@ class Decimal(NumericType):
     """
 
     __slots__ = ("precision", "scale")
-
     precision: int | None
     scale: int
 
@@ -558,6 +558,7 @@ class Categorical(DataType):
             or string value (lexical).
     """
 
+    __slots__ = ("ordering",)
     ordering: CategoricalOrdering | None
 
     def __init__(
@@ -592,6 +593,7 @@ class Enum(DataType):
         It may be changed at any point without it being considered a breaking change.
     """
 
+    __slots__ = ("categories",)
     categories: Series
 
     def __init__(self, categories: Series | Iterable[str]):

diff --git a/py-polars/polars/series/series.py b/py-polars/polars/series/series.py
@@ -237,7 +237,8 @@ class Series:
     ]
     """
 
-    _s: PySeries = None
+    __slots__ = ("_s",)
+    _s: PySeries
     _accessors: ClassVar[set[str]] = {
         "arr",
         "cat",