From 865a4b2f4070653823b589d3e08917a4634df6dc Mon Sep 17 00:00:00 2001
From: "Alinson S. Xavier" <git@axavier.org>
Date: Wed, 4 Aug 2021 11:34:56 -0500
Subject: [PATCH] Hdf5Sample: Store string vectors as "S" dtype instead of obj

---
 miplearn/features/sample.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/miplearn/features/sample.py b/miplearn/features/sample.py
index 917bb2b..0dd6005 100644
--- a/miplearn/features/sample.py
+++ b/miplearn/features/sample.py
@@ -247,8 +247,16 @@ class Hdf5Sample(Sample):
         if value is None:
             return
         self._assert_is_vector(value)
-        modified = [v if v is not None else "" for v in value]
-        self._put(key, modified, compress=True)
+
+        for v in value:
+            if isinstance(v, str):
+                value = np.array(
+                    [u if u is not None else b"" for u in value],
+                    dtype="S",
+                )
+                break
+
+        self._put(key, value, compress=True)
 
     @overrides
     def put_vector_list(self, key: str, value: VectorList) -> None: