def _pysave_arrays(args, dir=None): # @ReservedAssignment p = _tmp.mkdtemp(prefix='py4j-pyargs', dir=dir) n = 0 names = [] for arg in args: if isinstance(arg, ndarray): name = _path.join(p, "p%03d.npy" % n) _asave(name, arg) n += 1 names.append(name) return names
def _pysave_array(arg, dir=None, names=None): # @ReservedAssignment if not isinstance(arg, ndarray): return arg if not names: p = _tmp.mkdtemp(prefix='py4j-pyargs', dir=dir) n = 0 else: p = _path.dirname(names[-1]) n = len(names) name = _path.join(p, "p%03d.npy" % n) _asave(name, arg) return name