Add __slots__ to SubgraphView class

mggg · Jan 10, 2022 · 9c5265e · 9c5265e
1 parent 0251cab
commit 9c5265e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/gerrychain/partition/assignment.py b/gerrychain/partition/assignment.py
@@ -17,10 +17,10 @@ class Assignment(Mapping):
     An :class:`Assignment` has a ``parts`` property that is a dictionary of the form
     ``{part: <frozenset of nodes in part>}``.
     """
-    __slots__ = (
+    __slots__ = [
         'parts',
-        'mapping',
-    )
+        'mapping'
+    ]
 
     def __init__(self, parts, mapping=None, validate=True):
         if validate:

diff --git a/gerrychain/partition/subgraphs.py b/gerrychain/partition/subgraphs.py
@@ -1,4 +1,10 @@
 class SubgraphView:
+    __slots__ = [
+        "graph",
+        "parts",
+        "subgraphs_cache"
+    ]
+
     def __init__(self, graph, parts):
         self.graph = graph
         self.parts = parts