fix: ->jvm on object-dtype numpy arrays with np-array bindings (#187)

The np-array zero-copy bindings register a pyobject->jvm :ndarray method that
always maps the numpy dtype to a native tensor datatype. Object and other
non-numeric dtypes have no such mapping, so ->jvm threw "Unable to find
datatype: object" once the bindings were loaded. Route non-numeric dtypes to
the default element-wise copy conversion instead.

Author: jsavyasachi

src/libpython_clj2/python/np_array.clj

@@ -73,12 +73,25 @@
        :ctypes ctypes})))
 
 
+(defn- zero-copyable-dtype?
+  "Object/str/datetime arrays hold python-object pointers, not a numeric native
+  buffer, so only the numeric dtypes in py-dtype->dtype-map can become a tensor."
+  [pyobj]
+  (py-ffi/with-gil
+    (let [dtype-name (-> (py-proto/get-attr pyobj "dtype")
+                         (py-proto/get-attr "name"))]
+      (contains? py-dtype->dtype-map dtype-name))))
+
+
 (defmethod py-proto/pyobject->jvm :ndarray
   [pyobj opts]
-  (pygc/with-stack-context
-    (-> (numpy->desc pyobj)
-        (dtt/nd-buffer-descriptor->tensor)
-        (dtt/clone))))
+  (if (zero-copyable-dtype? pyobj)
+    (pygc/with-stack-context
+      (-> (numpy->desc pyobj)
+          (dtt/nd-buffer-descriptor->tensor)
+          (dtt/clone)))
+    ;; non-numeric dtype: element-wise copy via the default conversion (#187)
+    ((get-method py-proto/pyobject->jvm :default) pyobj opts)))
 
 
 (defmethod py-proto/pyobject-as-jvm :ndarray

test/libpython_clj2/numpy_test.clj

@@ -1,6 +1,8 @@
 (ns libpython-clj2.numpy-test
   (:require [clojure.test :refer [deftest is]]
             [libpython-clj2.python :as py]
+            ;; load the zero-copy bindings: this is the condition under which #187 fails
+            [libpython-clj2.python.np-array]
             [tech.v3.datatype :as dtype]
             [tech.v3.datatype.functional :as dfn]
             [tech.v3.tensor :as dtt]))
@@ -18,3 +20,11 @@
     (is (dfn/equals (dtt/ensure-tensor np-ary) tens))
     (is (dfn/equals [1 2 3 4]
                     (dtype/make-container :java-array :int64 np-ary)))))
+
+
+;; ->jvm on an object-dtype array with np-array bindings loaded (#187)
+(deftest object-dtype-ndarray->jvm
+  (let [empty-obj (py/call-attr-kw np-mod "array" [[]] {:dtype "object"})
+        mixed-obj (py/call-attr-kw np-mod "array" [["a" 1]] {:dtype "object"})]
+    (is (= [] (py/->jvm empty-obj)))
+    (is (= ["a" 1] (py/->jvm mixed-obj)))))