Run ruff check --fix on project

src/maggma/api/API.py

@@ -8,7 +8,6 @@
 from monty.json import MSONable
 from starlette.responses import RedirectResponse
 
-from maggma import __version__
 from maggma.api.resource import Resource
 
 

src/maggma/api/query_operator/sorting.py

@@ -1,6 +1,6 @@
-from typing import Optional, List
+from typing import Optional
 
-from fastapi import HTTPException, Query
+from fastapi import Query
 
 from maggma.api.query_operator import QueryOperator
 from maggma.api.utils import STORE_PARAMS

src/maggma/builders/projection_builder.py

@@ -1,7 +1,7 @@
 from copy import deepcopy
 from datetime import datetime
 from itertools import chain
-from typing import Dict, Iterable, List, Set, Union
+from typing import Dict, Iterable, List, Union
 
 from pydash import get
 

src/maggma/core/builder.py

@@ -5,7 +5,7 @@
 
 import logging
 from abc import ABCMeta, abstractmethod
-from typing import Any, Dict, Iterable, List, Optional, Union
+from typing import Any, Dict, Iterable, List, Union
 
 from monty.json import MontyDecoder, MSONable
 

src/maggma/core/store.py

@@ -7,7 +7,7 @@
 from abc import ABCMeta, abstractmethod, abstractproperty
 from datetime import datetime
 from enum import Enum
-from typing import Callable, Dict, Iterator, List, Optional, Tuple, Union
+from typing import Dict, Iterator, List, Optional, Tuple, Union
 
 from monty.dev import deprecated
 from monty.json import MontyDecoder, MSONable

src/maggma/stores/aws.py

@@ -8,7 +8,7 @@
 from concurrent.futures import wait
 from concurrent.futures.thread import ThreadPoolExecutor
 from hashlib import sha1
-from typing import Any, Dict, Iterator, List, Optional, Tuple, Union
+from typing import Dict, Iterator, List, Optional, Tuple, Union
 from json import dumps
 
 import msgpack  # type: ignore

src/maggma/stores/compound_stores.py

@@ -1,7 +1,7 @@
 """ Special stores that combine underlying Stores together """
 from datetime import datetime
 from itertools import groupby
-from typing import Any, Dict, Iterator, List, Optional, Tuple, Union
+from typing import Dict, Iterator, List, Optional, Tuple, Union
 
 from pydash import set_
 from pymongo import MongoClient

tests/api/test_aggregation_resource.py

@@ -1,8 +1,7 @@
 from random import randint
-from urllib.parse import urlencode
 
 import pytest
-from fastapi import FastAPI, Body
+from fastapi import FastAPI
 from pydantic import BaseModel, Field
 from starlette.testclient import TestClient
 

tests/api/test_query_operators.py

@@ -4,12 +4,11 @@
     NumericQuery,
     PaginationQuery,
     SparseFieldsQuery,
-    NumericQuery,
     SortQuery,
 )
 
 from pydantic import BaseModel, Field
-from fastapi import HTTPException, Query
+from fastapi import HTTPException
 from datetime import datetime
 
 from monty.serialization import loadfn, dumpfn

tests/api/test_submission_resource.py

@@ -1,16 +1,14 @@
 from random import randint
-from urllib.parse import urlencode
-from pydantic.utils import get_model
 import json
 import pytest
-from fastapi import FastAPI, Body
+from fastapi import FastAPI
 from pydantic import BaseModel, Field
 from starlette.testclient import TestClient
 
 from datetime import datetime
 from maggma.api.query_operator.core import QueryOperator
 
-from maggma.api.resource import SubmissionResource, post_resource
+from maggma.api.resource import SubmissionResource
 from maggma.api.query_operator import PaginationQuery
 
 from maggma.stores import MemoryStore

tests/cli/test_init.py

@@ -1,4 +1,3 @@
-import os
 import shutil
 from datetime import datetime
 from pathlib import Path

tests/cli/test_multiprocessing.py

@@ -1,4 +1,3 @@
-import asyncio
 import time
 from concurrent.futures import ThreadPoolExecutor
 

tests/stores/test_aws.py

@@ -1,12 +1,9 @@
 import time
-import zlib
 from datetime import datetime
 
 import boto3
-import msgpack
 import pytest
 from botocore.exceptions import ClientError
-from monty.msgpack import default, object_hook
 from moto import mock_s3
 
 from maggma.stores import MemoryStore, MongoStore, S3Store

tests/stores/test_shared_stores.py

@@ -182,7 +182,7 @@ def test_multistore_update(multistore, mongostore):
         mongostore_facade.update([large_doc, {"e": 1001}], key="e")
 
     mongostore_facade.safe_update = True
-    assert mongostore_facade.safe_update == True
+    assert mongostore_facade.safe_update is True
     mongostore_facade.update([large_doc, {"e": 1001}], key="e")
     assert mongostore_facade.query_one({"e": 1001}) is not None
 

tests/stores/test_ssh_tunnel.py

@@ -1,4 +1,3 @@
-import asyncio
 
 import paramiko
 import pymongo