Project import generated by Copybara.
GitOrigin-RevId: 6a704ded0bf489614797082e7e7cda1068477ef5
This commit is contained in:
@@ -397,7 +397,6 @@ void CalculatorGraphSubmodule(pybind11::module* module) {
|
||||
pybind11::function callback_fn) {
|
||||
RaisePyErrorIfNotOk(self->ObserveOutputStream(
|
||||
stream_name, [callback_fn, stream_name](const Packet& packet) {
|
||||
// Acquire a mutex so that only one callback_fn can run at once.
|
||||
absl::MutexLock lock(&callback_mutex);
|
||||
callback_fn(stream_name, packet);
|
||||
return absl::OkStatus();
|
||||
|
||||
@@ -87,6 +87,7 @@ class _PacketDataType(enum.Enum):
|
||||
BOOL_LIST = 'bool_list'
|
||||
INT = 'int'
|
||||
FLOAT = 'float'
|
||||
FLOAT_LIST = 'float_list'
|
||||
AUDIO = 'matrix'
|
||||
IMAGE = 'image_frame'
|
||||
PROTO = 'proto'
|
||||
@@ -108,6 +109,8 @@ NAME_TO_TYPE: Mapping[str, '_PacketDataType'] = {
|
||||
_PacketDataType.INT,
|
||||
'float':
|
||||
_PacketDataType.FLOAT,
|
||||
'::std::vector<float>':
|
||||
_PacketDataType.FLOAT_LIST,
|
||||
'::mediapipe::Matrix':
|
||||
_PacketDataType.AUDIO,
|
||||
'::mediapipe::ImageFrame':
|
||||
|
||||
Reference in New Issue
Block a user