Function ndarray_to_node

memory_graph/extension_numpy.py:11–18

(data, ndarray_data)

import memory_graph.config as config

def ndarray_to_node(data, ndarray_data):
    dim = len(ndarray_data.shape)
    if dim > 2:
        return Node_Linear(data, [i for i in ndarray_data])
    elif dim == 2:
        return Node_Table(data, ndarray_data)
    else:
        return Node_Linear(data, ndarray_data)

def get_numpy_immutables():
    import numpy as np