Skip to content

Commit

Permalink
remove atomate dependence in emmet-cli
Browse files Browse the repository at this point in the history
  • Loading branch information
esoteric-ephemera committed Apr 15, 2024
1 parent 75e83e2 commit 9b850c0
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 4 deletions.
6 changes: 3 additions & 3 deletions emmet-cli/emmet/cli/db.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@


class TaskStore:

_get_store_from_type: dict[str, Store] = {
"mongo": MongoStore,
"s3": S3Store,
Expand Down Expand Up @@ -225,7 +226,6 @@ def insert_object(self, *args, **kwargs) -> tuple[int, str]:
def insert_gridfs(
self,
dct: dict,
collection: str = "fs",
compression_type: Optional[Literal["zlib"]] = "zlib",
oid: Optional[ObjectId] = None,
task_id: Optional[Union[int, str]] = None,
Expand Down Expand Up @@ -268,12 +268,12 @@ def insert_maggma_store(
task_id: Optional[Any] = None,
) -> tuple[int, str]:
"""
Insert the given document into a Maggma store, first check if the store is already
Insert the given document into a Maggma store.
Args:
data: the document to be stored
collection (string): the name prefix for the maggma store
oid (str or ObjectId): the _id of the file; if specified, it must not already exist in GridFS
oid (str, ObjectId, None): the _id of the file; if specified, it must not already exist in GridFS
task_id(int or str): the task_id to store into the gridfs metadata
Returns:
file id, the type of compression used.
Expand Down
1 change: 0 additions & 1 deletion emmet-cli/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,6 @@
"click",
"colorama",
"mongogrant",
"atomate",
"mgzip",
"slurmpy",
"github3.py",
Expand Down

0 comments on commit 9b850c0

Please sign in to comment.