Boolean tensors support (#89)

* added support for dtype 'bool'

* added test cases for boolean tensors

* updated .gitignore file

* improved tests

* lint fix

Author: GuyAv46

.gitignore

@@ -2,10 +2,12 @@
 .pydevproject
 *.pyc
 .venv/
+venv/
 redisai.egg-info
 .idea
 .mypy_cache/
 build/
 dist/
 docs/_build/
 .DS_Store
+.vscode

redisai/utils.py

@@ -14,6 +14,7 @@
     "uint16": "UINT16",
     "uint32": "UINT32",
     "uint64": "UINT64",
+    "bool": "BOOL",
 }
 
 allowed_devices = {"CPU", "GPU"}

test/test.py

@@ -111,6 +111,12 @@ def test_set_non_numpy_tensor(self):
         self.assertEqual([2, 3, 4, 5], result["values"])
         self.assertEqual([2, 2], result["shape"])
 
+        con.tensorset("x", (1, 1, 0, 0), dtype="bool", shape=(2, 2))
+        result = con.tensorget("x", as_numpy=False)
+        self.assertEqual([True, True, False, False], result["values"])
+        self.assertEqual([2, 2], result["shape"])
+        self.assertEqual("BOOL", result["dtype"])
+
         with self.assertRaises(TypeError):
             con.tensorset("x", (2, 3, 4, 5), dtype="wrongtype", shape=(2, 2))
         con.tensorset("x", (2, 3, 4, 5), dtype="int8", shape=(2, 2))
@@ -144,6 +150,12 @@ def test_numpy_tensor(self):
         values = con.tensorget("x")
         self.assertEqual(values.dtype, np.float64)
 
+        input_array = np.array([True, False])
+        con.tensorset("x", input_array)
+        values = con.tensorget("x")
+        self.assertEqual(values.dtype, "bool")
+        self.assertTrue(np.array_equal(values, [True, False]))
+
         input_array = np.array([2, 3])
         con.tensorset("x", input_array)
         values = con.tensorget("x")