From fb2f0862e63b95c24f26def10bc76c069ef26ea9 Mon Sep 17 00:00:00 2001
From: SimonBoothroyd <simon.boothroyd@hotmail.com>
Date: Mon, 24 Jan 2022 12:33:22 +0000
Subject: [PATCH] Allow custom charge / WBO methods (#34)

---
 nagl/storage/_storage.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nagl/storage/_storage.py b/nagl/storage/_storage.py
index b347dd4..b66a2eb 100644
--- 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