gigl.src.inference.v1.lib.utils#
Functions
Gets the beam pipeline for running the inference dataflow job Args: gbml_config_pb_wrapper (GbmlConfigPbWrapper): GBML config wrapper for this inference run inference_blueprint (BaseInferenceBlueprint): Blueprint for running and saving inference for GBML pipelines applied_task_identifier (AppliedTaskIdentifier): Identifier for the GiGL job custom_worker_image_uri (Optional[str]): Uri to custom worker image node_type (NodeType): Node type being inferred uri_prefix_list (List[Uri]): List of prefixes for running inference for given node type temp_predictions_gcs_path (Optional[GcsUri]): Gcs uri for writing temp predictions temp_embeddings_gcs_path (Optional[GcsUri]): Gcs uri for writing temp embeddings Returns: pipeline (beam.Pipeline): Dataflow pipeline for running inference |
Classes