Serialising numpy ndarrays .tolist()

Author: Jason Costello

hypervector/resources/core/benchmark.py

@@ -1,3 +1,4 @@
+import numpy as np
 import requests
 
 import hypervector
@@ -47,12 +48,20 @@ def list(cls, ensemble):
     def new(cls, ensemble, expected_output):
         endpoint = f"{hypervector.API_BASE}/definition/{ensemble.definition_uuid}" \
                    f"/ensemble/{ensemble.ensemble_uuid}/benchmarks/add"
+
+        if isinstance(expected_output, np.ndarray):
+            expected_output = expected_output.tolist()
+
         data = {"expected_output": expected_output}
         response = requests.post(endpoint, json=data, headers=cls.get_headers()).json()
         return cls.from_response(response)
 
     def assert_equal(self, output_to_assert):
         endpoint = f"{hypervector.API_BASE}/benchmark/{self.benchmark_uuid}/assert"
+
+        if isinstance(output_to_assert, np.ndarray):
+            output_to_assert = output_to_assert.tolist()
+
         data = {"output_to_assert": output_to_assert}
         response = requests.post(endpoint, json=data, headers=self.get_headers()).json()
         return response

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name='hypervector-wrapper',
-    version='0.0.10',
+    version='0.0.11',
     description='Python wrapper to use the Hypervector API. Better data tests',
     long_description=long_description,
     long_description_content_type="text/markdown",
@@ -14,7 +14,8 @@
     url="https://github.com/hypervectorio/hypervector-wrapper",
     packages=find_packages(exclude=["tests"]),
     install_requires=[
-        "requests ~=2.25.0"
+        "requests ~=2.25.0",
+        "numpy ~=1.20.2"
     ],
     extras_require={
         "dev": [