def _is_instance_of(java_object: JavaObject,
java_type: Union[JavaObject, JavaClass]) -> bool:
if isinstance(java_type, JavaObject):
return java_object.equals(java_type)
elif isinstance(java_type, JavaClass):
return java_object.getClass().isAssignableFrom(
java_type._java_lang_class)
return False
def from_java(cls, java_value_info: JavaObject) -> "ValueInfo":
className = java_value_info.getClass().getName()
if className == "ai.onnxruntime.TensorInfo":
return TensorInfo.from_java(java_value_info)
elif className == "ai.onnxruntime.MapInfo":
return MapInfo.from_java(java_value_info)
else:
return SequenceInfo.from_java(java_value_info)
def wrap_transformer(j_transformer: JavaObject):
"""
Wrap a Java instance of `Transformer` to a Python object of corresponding class.
:param j_transformer: a Java instance of `Transformer`.
:return: a Python object with corresponding class.
"""
model_name = j_transformer.getClass().getSimpleName()
import importlib
model_cls = getattr(importlib.import_module("pyalink.alink.pipeline"),
model_name)
return model_cls(j_transformer=j_transformer)
def wrap_model(j_model: JavaObject):
model_name = j_model.getClass().getSimpleName()
import importlib
model_cls = getattr(importlib.import_module("pyalink.alink.pipeline"), model_name)
return model_cls(j_model=j_model)
def wrap_transformer(j_transformer: JavaObject):
model_name = j_transformer.getClass().getSimpleName()
import importlib
model_cls = getattr(importlib.import_module("pyalink.alink.pipeline"), model_name)
return model_cls(j_transformer=j_transformer)
