Allow custom charge / WBO methods (#34)

SimonBoothroyd · Jan 24, 2022 · fb2f086 · fb2f086
1 parent 9b6ecb8
commit fb2f086
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nagl/storage/_storage.py b/nagl/storage/_storage.py
@@ -48,8 +48,8 @@
 _logger = logging.getLogger(__name__)
 
 
-ChargeMethod = Literal["am1", "am1bcc"]
-WBOMethod = Literal["am1"]
+ChargeMethod = Union[Literal["am1", "am1bcc"], str]
+WBOMethod = Union[Literal["am1"], str]
 
 DBQueryResult = Tuple[
     int,  # DBMoleculeRecord.id