Set type hint

materialsproject · Sep 22, 2023 · 80195ec · 80195ec
1 parent 95bc61f
commit 80195ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/maggma/utils.py b/src/maggma/utils.py
@@ -7,7 +7,7 @@
 import uuid
 from datetime import datetime, timedelta
 from importlib import import_module
-from typing import Dict, Iterable, Optional, Union
+from typing import Dict, Iterable, Optional, Union, Set
 
 from bson.json_util import ObjectId
 from dateutil import parser
@@ -264,7 +264,7 @@ def emit(self, record):
             maggma_record["build_id"] = self.build_id
             self.reporting_store.update(maggma_record, key="_id")
 
-def get_flat_models_from_model(model: BaseModel, known_models: set[BaseModel] = set()):
+def get_flat_models_from_model(model: BaseModel, known_models: Set[BaseModel] = set()):
     """Get all sub-models from a pydantic model.
 
     Args: