cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow ModelResourcesCalculator to use file_pointer_meta.

Browse Source 
PiperOrigin-RevId: 517742722
This commit is contained in:
Jiuqiang Tang 2023-03-19 01:31:07 -07:00 committed by Copybara-Service
parent 2e5ea174fe
commit f06e4224b8
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mediapipe/tasks/cc/core/model_resources_calculator.cc
View File

@ -77,9 +77,11 @@ class ModelResourcesCalculator : public api2::Node {
if (options.has_model_file()) {
RET_CHECK(options.model_file().has_file_content() ||
options.model_file().has_file_descriptor_meta() ||
options.model_file().has_file_name())
options.model_file().has_file_name() ||
options.model_file().has_file_pointer_meta())
<< "'model_file' must specify at least one of "
"'file_content', 'file_descriptor_meta', or 'file_name'";
"'file_content', 'file_descriptor_meta', 'file_name', or "
"'file_pointer_meta'";
}
return absl::OkStatus();
}

6
mediapipe/tasks/cc/core/model_resources_calculator_test.cc
View File

@ -179,9 +179,9 @@ TEST_F(ModelResourcesCalculatorTest, EmptyExternalFileProto) {
auto status = graph.Initialize(graph_config);
ASSERT_FALSE(status.ok());
EXPECT_THAT(status.message(),
testing::HasSubstr(
"'model_file' must specify at least one of "
"'file_content', 'file_descriptor_meta', or 'file_name'"));
testing::HasSubstr("'model_file' must specify at least one of "
"'file_content', 'file_descriptor_meta', "
"'file_name', or 'file_pointer_meta'"));
}
TEST_F(ModelResourcesCalculatorTest, GraphServiceNotAvailable) {