diff --git a/src/maggma/stores/advanced_stores.py b/src/maggma/stores/advanced_stores.py index 05d3e9d48..58959a01f 100644 --- a/src/maggma/stores/advanced_stores.py +++ b/src/maggma/stores/advanced_stores.py @@ -327,7 +327,12 @@ def groupby( lazy_substitute(criteria, self.reverse_aliases) return self.store.groupby( - keys=keys, properties=properties, criteria=criteria, skip=skip, limit=limit, sort=sort, + keys=keys, + properties=properties, + criteria=criteria, + skip=skip, + limit=limit, + sort=sort, ) def update(self, docs: Union[List[Dict], Dict], key: Union[List, str, None] = None): @@ -502,7 +507,12 @@ def groupby( ) return self.store.groupby( - keys=keys, properties=properties, criteria=criteria, skip=skip, limit=limit, sort=sort + keys=keys, + properties=properties, + criteria=criteria, + skip=skip, + limit=limit, + sort=sort, ) def update(self, docs: Union[List[Dict], Dict], key: Union[List, str, None] = None): diff --git a/src/maggma/stores/mongolike.py b/src/maggma/stores/mongolike.py index 1c7b78bbc..a49d91ad0 100644 --- a/src/maggma/stores/mongolike.py +++ b/src/maggma/stores/mongolike.py @@ -7,7 +7,7 @@ from pathlib import Path import yaml -from itertools import chain, groupby +from itertools import chain from socket import socket import warnings from typing import Any, Dict, Iterator, List, Optional, Tuple, Union