Minor ONNX Wrapper Fix

bookbot-kids · Sep 27, 2024 · 3733c32 · 3733c32
1 parent 3ae05ad
commit 3733c32
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/g2p_id/__init__.py b/g2p_id/__init__.py
@@ -20,5 +20,5 @@
 from .onnx_utils import WrapInferenceSession
 from .text_processor import TextProcessor
 
-__version__ = "0.3.6"
+__version__ = "0.3.6.1"
 __all__ = ["G2p", "LSTM", "BERT", "WrapInferenceSession", "TextProcessor"]
diff --git a/g2p_id/onnx_utils.py b/g2p_id/onnx_utils.py
@@ -40,5 +40,5 @@ def __getstate__(self):
 
     def __setstate__(self, values):
         self.onnx_bytes = values["onnx_bytes"]
-        self.providers = values["providers"]
+        self.providers = values.get("providers", None)
         self.sess = ort.InferenceSession(self.onnx_bytes, self.providers)
diff --git a/setup.py b/setup.py
@@ -14,7 +14,7 @@
 if __name__ == "__main__":
     setup(
         name="g2p_id_py",
-        version="0.3.6",
+        version="0.3.6.1",
         description="Indonesian G2P.",
         long_description=long_description,
         long_description_content_type="text/markdown",