diff --git a/mediapipe/tasks/cc/core/model_resources_calculator.cc b/mediapipe/tasks/cc/core/model_resources_calculator.cc index 72a7b33a3..d5c8cd502 100644 --- a/mediapipe/tasks/cc/core/model_resources_calculator.cc +++ b/mediapipe/tasks/cc/core/model_resources_calculator.cc @@ -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(); } diff --git a/mediapipe/tasks/cc/core/model_resources_calculator_test.cc b/mediapipe/tasks/cc/core/model_resources_calculator_test.cc index 58659c77d..83134a8c7 100644 --- a/mediapipe/tasks/cc/core/model_resources_calculator_test.cc +++ b/mediapipe/tasks/cc/core/model_resources_calculator_test.cc @@ -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) {