gigl.src.common.modeling_task_specs.utils.infer_task_inputs

gigl.src.common.modeling_task_specs.utils.infer.infer_task_inputs(model: DistributedDataParallel | Module, gbml_config_pb_wrapper: GbmlConfigPbWrapper, main_batch: NodeAnchorBasedLinkPredictionBatch, random_neg_batch: RootedNodeNeighborhoodBatch, should_eval: bool, device: device) → NodeAnchorBasedLinkPredictionTaskInputs