Use cached_statepoint in groupby.

glotzerlab · Feb 12, 2024 · 1de7155 · 1de7155
1 parent 4ebf74f
commit 1de7155
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/signac/project.py b/signac/project.py
@@ -1913,7 +1913,7 @@ def keyfunction(job):
                 else:
 
                     def keyfunction(job):
-                        return job.sp[stripped_key]
+                        return job.cached_statepoint[stripped_key]
 
             else:
                 if _is_doc_key(key):
@@ -1924,7 +1924,7 @@ def keyfunction(job):
                 else:
 
                     def keyfunction(job):
-                        return job.sp.get(stripped_key, default)
+                        return job.cached_statepoint.get(stripped_key, default)
 
         elif isinstance(key, Iterable):
             sp_keys = []
@@ -1943,15 +1943,15 @@ def keyfunction(job):
 
                 def keyfunction(job):
                     return tuple(
-                        [job.sp[k] for k in sp_keys]
+                        [job.cached_statepoint[k] for k in sp_keys]
                         + [job.document[k] for k in doc_keys]
                     )
 
             else:
 
                 def keyfunction(job):
                     return tuple(
-                        [job.sp.get(k, default) for k in sp_keys]
+                        [job.cached_statepoint.get(k, default) for k in sp_keys]
                         + [job.document.get(k, default) for k in doc_keys]
                     )