File python_callbacks.h

struct PyCallbackIOWriter : public faiss::IOWriter

#include <python_callbacks.h>

Copyright (c) Facebook, Inc. and its affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Public Functions

explicit PyCallbackIOWriter(PyObject *callback, size_t bs = 1024 * 1024)

Callback: Python function that takes a bytes object and returns the number of bytes successfully written.

virtual size_t operator()(const void *ptrv, size_t size, size_t nitems) override

~PyCallbackIOWriter() override

Public Members

PyObject *callback

size_t bs

struct PyCallbackIOReader : public faiss::IOReader

Public Functions

explicit PyCallbackIOReader(PyObject *callback, size_t bs = 1024 * 1024)

Callback: Python function that takes a size and returns a bytes object with the resulting read

virtual size_t operator()(void *ptrv, size_t size, size_t nitems) override

~PyCallbackIOReader() override

Public Members

PyObject *callback

size_t bs

struct PyCallbackIDSelector : public faiss::IDSelector

Public Functions

explicit PyCallbackIDSelector(PyObject *callback)

virtual bool is_member(faiss::idx_t id) const override

~PyCallbackIDSelector() override

Public Members

PyObject *callback