diff --git a/proto/snapchat/research/gbml/gigl_resource_config.proto b/proto/snapchat/research/gbml/gigl_resource_config.proto
index a45767258..bab99bd27 100644
--- a/proto/snapchat/research/gbml/gigl_resource_config.proto
+++ b/proto/snapchat/research/gbml/gigl_resource_config.proto
@@ -54,7 +54,7 @@ message VertexAiTrainerConfig {
   // Num workers for training job
   uint32 num_replicas = 4;
  }
- 
+
  // (deprecated)
  // Configuration for KFP training resources
  message KFPTrainerConfig {
@@ -86,11 +86,17 @@ message VertexAiResourceConfig {
   uint32 gpu_limit = 3;
   // Num workers for job
   uint32 num_replicas = 4;
-  // Timeout in seconds for the job. If unset or zero, will use the default @ google.cloud.aiplatform.CustomJob, which is 7 days: 
+  // Timeout in seconds for the job. If unset or zero, will use the default @ google.cloud.aiplatform.CustomJob, which is 7 days:
   // https://github.com/googleapis/python-aiplatform/blob/58fbabdeeefd1ccf1a9d0c22eeb5606aeb9c2266/google/cloud/aiplatform/jobs.py#L2252-L2253
   uint32 timeout = 5;
+  // Region override
+  // If provided, then the Vertex AI Job will be launched in the provided region.
+  // Otherwise, will launch jobs in the region specified at CommonComputeConfig.region
+  // ex: "us-west1"
+  // NOTE: If set, then there may be data egress costs from CommonComputeConfig.region -> gcp_region_override
+  string gcp_region_override = 6;
  }
- 
+
  // Configuration for KFP job resources
  message KFPResourceConfig {
   // Num CPU requested for job (str) which can be a number or a number followed by "m", which means 1/1000
@@ -187,7 +193,7 @@ message GiglResourceConfig {
  // (deprecated)
  // Configuration for inferencer
  DataflowResourceConfig inferencer_config = 16 [deprecated=true];
- 
+
  // Configuration for distributed trainer
  TrainerResourceConfig trainer_resource_config = 17;
  // Configuration for distributed inferencer
diff --git a/python/gigl/src/common/types/pb_wrappers/gigl_resource_config.py b/python/gigl/src/common/types/pb_wrappers/gigl_resource_config.py
index d454a8b5a..76c339e84 100644
--- a/python/gigl/src/common/types/pb_wrappers/gigl_resource_config.py
+++ b/python/gigl/src/common/types/pb_wrappers/gigl_resource_config.py
@@ -230,6 +230,42 @@ def region(self) -> str:
         """
         return self.shared_resource_config.common_compute_config.region
 
+    @property
+    def vertex_ai_trainer_region(self) -> str:
+        """
+        Returns the region specified in the resource config for Vertex AI trainer. (e.g. us-central1)
+        By default this will be set to shared_resource_config.common_compute_config.region
+        But if trainer_resource_config.vertex_ai_trainer_config.vertex_ai_trainer_config.gcp_region_override is set, it will return that instead.
+        Raises an error if the trainer is *not* a Vertex AI trainer.
+        """
+        trainer_config = self.trainer_config
+        if not isinstance(trainer_config, VertexAiResourceConfig):
+            raise ValueError(
+                "Vertex AI trainer region is only supported for Vertex AI trainers."
+            )
+        if trainer_config.gcp_region_override:
+            return trainer_config.gcp_region_override
+        else:
+            return self.region
+
+    @property
+    def vertex_ai_inferencer_region(self) -> str:
+        """
+        Returns the region specified in the resource config for Vertex AI inferencer. (e.g. us-central1)
+        By default this will be set to shared_resource_config.common_compute_config.region
+        But if inferencer_resource_config.vertex_ai_inferencer_config.vertex_ai_inferencer_config.gcp_region_override is set, it will return that instead.
+        Raises an error if the inferencer is *not* a Vertex AI inferencer.
+        """
+        inferencer_config = self.inferencer_config
+        if not isinstance(inferencer_config, VertexAiResourceConfig):
+            raise ValueError(
+                "Vertex AI inferencer region is only supported for Vertex AI inferencers."
+            )
+        if inferencer_config.gcp_region_override:
+            return inferencer_config.gcp_region_override
+        else:
+            return self.region
+
     @property
     def trainer_config(
         self,
diff --git a/python/gigl/src/inference/v2/glt_inferencer.py b/python/gigl/src/inference/v2/glt_inferencer.py
index 1d5ded7b2..fdda96d49 100644
--- a/python/gigl/src/inference/v2/glt_inferencer.py
+++ b/python/gigl/src/inference/v2/glt_inferencer.py
@@ -122,9 +122,10 @@ def __execute_VAI_inference(
             if inferencer_resource_config.timeout
             else None,
         )
+
         vertex_ai_service = VertexAIService(
             project=resource_config_wrapper.project,
-            location=resource_config_wrapper.region,
+            location=resource_config_wrapper.vertex_ai_inferencer_region,
             service_account=resource_config_wrapper.service_account_email,
             staging_bucket=resource_config_wrapper.temp_assets_regional_bucket_path.uri,
         )
diff --git a/python/gigl/src/training/v2/glt_trainer.py b/python/gigl/src/training/v2/glt_trainer.py
index 3093a3f88..9d2aea4c9 100644
--- a/python/gigl/src/training/v2/glt_trainer.py
+++ b/python/gigl/src/training/v2/glt_trainer.py
@@ -120,9 +120,10 @@ def __execute_VAI_training(
             if trainer_resource_config.timeout
             else None,
         )
+
         vertex_ai_service = VertexAIService(
             project=resource_config.project,
-            location=resource_config.region,
+            location=resource_config.vertex_ai_trainer_region,
             service_account=resource_config.service_account_email,
             staging_bucket=resource_config.temp_assets_regional_bucket_path.uri,
         )
@@ -205,8 +206,8 @@ def run(
 
     initialize_metrics(task_config_uri=task_config_uri, service_name=args.job_name)
 
-    glt_inferencer = GLTTrainer()
-    glt_inferencer.run(
+    glt_trainer = GLTTrainer()
+    glt_trainer.run(
         applied_task_identifier=applied_task_identifier,
         task_config_uri=task_config_uri,
         resource_config_uri=resource_config_uri,
diff --git a/python/snapchat/research/gbml/gbml_config_pb2.pyi b/python/snapchat/research/gbml/gbml_config_pb2.pyi
index ba532b202..3ce465beb 100644
--- a/python/snapchat/research/gbml/gbml_config_pb2.pyi
+++ b/python/snapchat/research/gbml/gbml_config_pb2.pyi
@@ -445,7 +445,7 @@ class GbmlConfig(google.protobuf.message.Message):
         GRAPH_DB_ARGS_FIELD_NUMBER: builtins.int
         GRAPH_DB_SAMPLER_CONFIG_FIELD_NUMBER: builtins.int
         graph_db_ingestion_cls_path: builtins.str
-        """Python class path pointing to user-written 
+        """Python class path pointing to user-written
         `BaseIngestion`` class definition. e.g. `my.team.graph_db.BaseInjectionImpl`.
         This class is currently, as an implementation detail, used for injestion only.
         We document this *purely* for information purposes and may change the implementation at any time.
diff --git a/python/snapchat/research/gbml/gigl_resource_config_pb2.py b/python/snapchat/research/gbml/gigl_resource_config_pb2.py
index d23e7759e..bef644bfb 100644
--- a/python/snapchat/research/gbml/gigl_resource_config_pb2.py
+++ b/python/snapchat/research/gbml/gigl_resource_config_pb2.py
@@ -15,7 +15,7 @@
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1snapchat/research/gbml/gigl_resource_config.proto\x12\x16snapchat.research.gbml\"Y\n\x13SparkResourceConfig\x12\x14\n\x0cmachine_type\x18\x01 \x01(\t\x12\x16\n\x0enum_local_ssds\x18\x02 \x01(\r\x12\x14\n\x0cnum_replicas\x18\x03 \x01(\r\"r\n\x16\x44\x61taflowResourceConfig\x12\x13\n\x0bnum_workers\x18\x01 \x01(\r\x12\x17\n\x0fmax_num_workers\x18\x02 \x01(\r\x12\x14\n\x0cmachine_type\x18\x03 \x01(\t\x12\x14\n\x0c\x64isk_size_gb\x18\x04 \x01(\r\"\xbc\x01\n\x16\x44\x61taPreprocessorConfig\x12P\n\x18\x65\x64ge_preprocessor_config\x18\x01 \x01(\x0b\x32..snapchat.research.gbml.DataflowResourceConfig\x12P\n\x18node_preprocessor_config\x18\x02 \x01(\x0b\x32..snapchat.research.gbml.DataflowResourceConfig\"h\n\x15VertexAiTrainerConfig\x12\x14\n\x0cmachine_type\x18\x01 \x01(\t\x12\x10\n\x08gpu_type\x18\x02 \x01(\t\x12\x11\n\tgpu_limit\x18\x03 \x01(\r\x12\x14\n\x0cnum_replicas\x18\x04 \x01(\r\"z\n\x10KFPTrainerConfig\x12\x13\n\x0b\x63pu_request\x18\x01 \x01(\t\x12\x16\n\x0ememory_request\x18\x02 \x01(\t\x12\x10\n\x08gpu_type\x18\x03 \x01(\t\x12\x11\n\tgpu_limit\x18\x04 \x01(\r\x12\x14\n\x0cnum_replicas\x18\x05 \x01(\r\")\n\x12LocalTrainerConfig\x12\x13\n\x0bnum_workers\x18\x01 \x01(\r\"z\n\x16VertexAiResourceConfig\x12\x14\n\x0cmachine_type\x18\x01 \x01(\t\x12\x10\n\x08gpu_type\x18\x02 \x01(\t\x12\x11\n\tgpu_limit\x18\x03 \x01(\r\x12\x14\n\x0cnum_replicas\x18\x04 \x01(\r\x12\x0f\n\x07timeout\x18\x05 \x01(\r\"{\n\x11KFPResourceConfig\x12\x13\n\x0b\x63pu_request\x18\x01 \x01(\t\x12\x16\n\x0ememory_request\x18\x02 \x01(\t\x12\x10\n\x08gpu_type\x18\x03 \x01(\t\x12\x11\n\tgpu_limit\x18\x04 \x01(\r\x12\x14\n\x0cnum_replicas\x18\x05 \x01(\r\"*\n\x13LocalResourceConfig\x12\x13\n\x0bnum_workers\x18\x01 \x01(\r\"\x93\x02\n\x18\x44istributedTrainerConfig\x12Q\n\x18vertex_ai_trainer_config\x18\x01 \x01(\x0b\x32-.snapchat.research.gbml.VertexAiTrainerConfigH\x00\x12\x46\n\x12kfp_trainer_config\x18\x02 \x01(\x0b\x32(.snapchat.research.gbml.KFPTrainerConfigH\x00\x12J\n\x14local_trainer_config\x18\x03 \x01(\x0b\x32*.snapchat.research.gbml.LocalTrainerConfigH\x00\x42\x10\n\x0etrainer_config\"\x93\x02\n\x15TrainerResourceConfig\x12R\n\x18vertex_ai_trainer_config\x18\x01 \x01(\x0b\x32..snapchat.research.gbml.VertexAiResourceConfigH\x00\x12G\n\x12kfp_trainer_config\x18\x02 \x01(\x0b\x32).snapchat.research.gbml.KFPResourceConfigH\x00\x12K\n\x14local_trainer_config\x18\x03 \x01(\x0b\x32+.snapchat.research.gbml.LocalResourceConfigH\x00\x42\x10\n\x0etrainer_config\"\xac\x02\n\x18InferencerResourceConfig\x12U\n\x1bvertex_ai_inferencer_config\x18\x01 \x01(\x0b\x32..snapchat.research.gbml.VertexAiResourceConfigH\x00\x12T\n\x1a\x64\x61taflow_inferencer_config\x18\x02 \x01(\x0b\x32..snapchat.research.gbml.DataflowResourceConfigH\x00\x12N\n\x17local_inferencer_config\x18\x03 \x01(\x0b\x32+.snapchat.research.gbml.LocalResourceConfigH\x00\x42\x13\n\x11inferencer_config\"\xa3\x04\n\x14SharedResourceConfig\x12Y\n\x0fresource_labels\x18\x01 \x03(\x0b\x32@.snapchat.research.gbml.SharedResourceConfig.ResourceLabelsEntry\x12_\n\x15\x63ommon_compute_config\x18\x02 \x01(\x0b\x32@.snapchat.research.gbml.SharedResourceConfig.CommonComputeConfig\x1a\x97\x02\n\x13\x43ommonComputeConfig\x12\x0f\n\x07project\x18\x01 \x01(\t\x12\x0e\n\x06region\x18\x02 \x01(\t\x12\x1a\n\x12temp_assets_bucket\x18\x03 \x01(\t\x12#\n\x1btemp_regional_assets_bucket\x18\x04 \x01(\t\x12\x1a\n\x12perm_assets_bucket\x18\x05 \x01(\t\x12#\n\x1btemp_assets_bq_dataset_name\x18\x06 \x01(\t\x12!\n\x19\x65mbedding_bq_dataset_name\x18\x07 \x01(\t\x12!\n\x19gcp_service_account_email\x18\x08 \x01(\t\x12\x17\n\x0f\x64\x61taflow_runner\x18\x0b \x01(\t\x1a\x35\n\x13ResourceLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xc8\x05\n\x12GiglResourceConfig\x12$\n\x1ashared_resource_config_uri\x18\x01 \x01(\tH\x00\x12N\n\x16shared_resource_config\x18\x02 \x01(\x0b\x32,.snapchat.research.gbml.SharedResourceConfigH\x00\x12K\n\x13preprocessor_config\x18\x0c \x01(\x0b\x32..snapchat.research.gbml.DataPreprocessorConfig\x12L\n\x17subgraph_sampler_config\x18\r \x01(\x0b\x32+.snapchat.research.gbml.SparkResourceConfig\x12K\n\x16split_generator_config\x18\x0e \x01(\x0b\x32+.snapchat.research.gbml.SparkResourceConfig\x12L\n\x0etrainer_config\x18\x0f \x01(\x0b\x32\x30.snapchat.research.gbml.DistributedTrainerConfigB\x02\x18\x01\x12M\n\x11inferencer_config\x18\x10 \x01(\x0b\x32..snapchat.research.gbml.DataflowResourceConfigB\x02\x18\x01\x12N\n\x17trainer_resource_config\x18\x11 \x01(\x0b\x32-.snapchat.research.gbml.TrainerResourceConfig\x12T\n\x1ainferencer_resource_config\x18\x12 \x01(\x0b\x32\x30.snapchat.research.gbml.InferencerResourceConfigB\x11\n\x0fshared_resource*\xf3\x01\n\tComponent\x12\x15\n\x11\x43omponent_Unknown\x10\x00\x12\x1e\n\x1a\x43omponent_Config_Validator\x10\x01\x12\x1e\n\x1a\x43omponent_Config_Populator\x10\x02\x12\x1f\n\x1b\x43omponent_Data_Preprocessor\x10\x03\x12\x1e\n\x1a\x43omponent_Subgraph_Sampler\x10\x04\x12\x1d\n\x19\x43omponent_Split_Generator\x10\x05\x12\x15\n\x11\x43omponent_Trainer\x10\x06\x12\x18\n\x14\x43omponent_Inferencer\x10\x07\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1snapchat/research/gbml/gigl_resource_config.proto\x12\x16snapchat.research.gbml\"Y\n\x13SparkResourceConfig\x12\x14\n\x0cmachine_type\x18\x01 \x01(\t\x12\x16\n\x0enum_local_ssds\x18\x02 \x01(\r\x12\x14\n\x0cnum_replicas\x18\x03 \x01(\r\"r\n\x16\x44\x61taflowResourceConfig\x12\x13\n\x0bnum_workers\x18\x01 \x01(\r\x12\x17\n\x0fmax_num_workers\x18\x02 \x01(\r\x12\x14\n\x0cmachine_type\x18\x03 \x01(\t\x12\x14\n\x0c\x64isk_size_gb\x18\x04 \x01(\r\"\xbc\x01\n\x16\x44\x61taPreprocessorConfig\x12P\n\x18\x65\x64ge_preprocessor_config\x18\x01 \x01(\x0b\x32..snapchat.research.gbml.DataflowResourceConfig\x12P\n\x18node_preprocessor_config\x18\x02 \x01(\x0b\x32..snapchat.research.gbml.DataflowResourceConfig\"h\n\x15VertexAiTrainerConfig\x12\x14\n\x0cmachine_type\x18\x01 \x01(\t\x12\x10\n\x08gpu_type\x18\x02 \x01(\t\x12\x11\n\tgpu_limit\x18\x03 \x01(\r\x12\x14\n\x0cnum_replicas\x18\x04 \x01(\r\"z\n\x10KFPTrainerConfig\x12\x13\n\x0b\x63pu_request\x18\x01 \x01(\t\x12\x16\n\x0ememory_request\x18\x02 \x01(\t\x12\x10\n\x08gpu_type\x18\x03 \x01(\t\x12\x11\n\tgpu_limit\x18\x04 \x01(\r\x12\x14\n\x0cnum_replicas\x18\x05 \x01(\r\")\n\x12LocalTrainerConfig\x12\x13\n\x0bnum_workers\x18\x01 \x01(\r\"\x97\x01\n\x16VertexAiResourceConfig\x12\x14\n\x0cmachine_type\x18\x01 \x01(\t\x12\x10\n\x08gpu_type\x18\x02 \x01(\t\x12\x11\n\tgpu_limit\x18\x03 \x01(\r\x12\x14\n\x0cnum_replicas\x18\x04 \x01(\r\x12\x0f\n\x07timeout\x18\x05 \x01(\r\x12\x1b\n\x13gcp_region_override\x18\x06 \x01(\t\"{\n\x11KFPResourceConfig\x12\x13\n\x0b\x63pu_request\x18\x01 \x01(\t\x12\x16\n\x0ememory_request\x18\x02 \x01(\t\x12\x10\n\x08gpu_type\x18\x03 \x01(\t\x12\x11\n\tgpu_limit\x18\x04 \x01(\r\x12\x14\n\x0cnum_replicas\x18\x05 \x01(\r\"*\n\x13LocalResourceConfig\x12\x13\n\x0bnum_workers\x18\x01 \x01(\r\"\x93\x02\n\x18\x44istributedTrainerConfig\x12Q\n\x18vertex_ai_trainer_config\x18\x01 \x01(\x0b\x32-.snapchat.research.gbml.VertexAiTrainerConfigH\x00\x12\x46\n\x12kfp_trainer_config\x18\x02 \x01(\x0b\x32(.snapchat.research.gbml.KFPTrainerConfigH\x00\x12J\n\x14local_trainer_config\x18\x03 \x01(\x0b\x32*.snapchat.research.gbml.LocalTrainerConfigH\x00\x42\x10\n\x0etrainer_config\"\x93\x02\n\x15TrainerResourceConfig\x12R\n\x18vertex_ai_trainer_config\x18\x01 \x01(\x0b\x32..snapchat.research.gbml.VertexAiResourceConfigH\x00\x12G\n\x12kfp_trainer_config\x18\x02 \x01(\x0b\x32).snapchat.research.gbml.KFPResourceConfigH\x00\x12K\n\x14local_trainer_config\x18\x03 \x01(\x0b\x32+.snapchat.research.gbml.LocalResourceConfigH\x00\x42\x10\n\x0etrainer_config\"\xac\x02\n\x18InferencerResourceConfig\x12U\n\x1bvertex_ai_inferencer_config\x18\x01 \x01(\x0b\x32..snapchat.research.gbml.VertexAiResourceConfigH\x00\x12T\n\x1a\x64\x61taflow_inferencer_config\x18\x02 \x01(\x0b\x32..snapchat.research.gbml.DataflowResourceConfigH\x00\x12N\n\x17local_inferencer_config\x18\x03 \x01(\x0b\x32+.snapchat.research.gbml.LocalResourceConfigH\x00\x42\x13\n\x11inferencer_config\"\xa3\x04\n\x14SharedResourceConfig\x12Y\n\x0fresource_labels\x18\x01 \x03(\x0b\x32@.snapchat.research.gbml.SharedResourceConfig.ResourceLabelsEntry\x12_\n\x15\x63ommon_compute_config\x18\x02 \x01(\x0b\x32@.snapchat.research.gbml.SharedResourceConfig.CommonComputeConfig\x1a\x97\x02\n\x13\x43ommonComputeConfig\x12\x0f\n\x07project\x18\x01 \x01(\t\x12\x0e\n\x06region\x18\x02 \x01(\t\x12\x1a\n\x12temp_assets_bucket\x18\x03 \x01(\t\x12#\n\x1btemp_regional_assets_bucket\x18\x04 \x01(\t\x12\x1a\n\x12perm_assets_bucket\x18\x05 \x01(\t\x12#\n\x1btemp_assets_bq_dataset_name\x18\x06 \x01(\t\x12!\n\x19\x65mbedding_bq_dataset_name\x18\x07 \x01(\t\x12!\n\x19gcp_service_account_email\x18\x08 \x01(\t\x12\x17\n\x0f\x64\x61taflow_runner\x18\x0b \x01(\t\x1a\x35\n\x13ResourceLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xc8\x05\n\x12GiglResourceConfig\x12$\n\x1ashared_resource_config_uri\x18\x01 \x01(\tH\x00\x12N\n\x16shared_resource_config\x18\x02 \x01(\x0b\x32,.snapchat.research.gbml.SharedResourceConfigH\x00\x12K\n\x13preprocessor_config\x18\x0c \x01(\x0b\x32..snapchat.research.gbml.DataPreprocessorConfig\x12L\n\x17subgraph_sampler_config\x18\r \x01(\x0b\x32+.snapchat.research.gbml.SparkResourceConfig\x12K\n\x16split_generator_config\x18\x0e \x01(\x0b\x32+.snapchat.research.gbml.SparkResourceConfig\x12L\n\x0etrainer_config\x18\x0f \x01(\x0b\x32\x30.snapchat.research.gbml.DistributedTrainerConfigB\x02\x18\x01\x12M\n\x11inferencer_config\x18\x10 \x01(\x0b\x32..snapchat.research.gbml.DataflowResourceConfigB\x02\x18\x01\x12N\n\x17trainer_resource_config\x18\x11 \x01(\x0b\x32-.snapchat.research.gbml.TrainerResourceConfig\x12T\n\x1ainferencer_resource_config\x18\x12 \x01(\x0b\x32\x30.snapchat.research.gbml.InferencerResourceConfigB\x11\n\x0fshared_resource*\xf3\x01\n\tComponent\x12\x15\n\x11\x43omponent_Unknown\x10\x00\x12\x1e\n\x1a\x43omponent_Config_Validator\x10\x01\x12\x1e\n\x1a\x43omponent_Config_Populator\x10\x02\x12\x1f\n\x1b\x43omponent_Data_Preprocessor\x10\x03\x12\x1e\n\x1a\x43omponent_Subgraph_Sampler\x10\x04\x12\x1d\n\x19\x43omponent_Split_Generator\x10\x05\x12\x15\n\x11\x43omponent_Trainer\x10\x06\x12\x18\n\x14\x43omponent_Inferencer\x10\x07\x62\x06proto3')
 
 _COMPONENT = DESCRIPTOR.enum_types_by_name['Component']
 Component = enum_type_wrapper.EnumTypeWrapper(_COMPONENT)
@@ -168,8 +168,8 @@
   _GIGLRESOURCECONFIG.fields_by_name['trainer_config']._serialized_options = b'\030\001'
   _GIGLRESOURCECONFIG.fields_by_name['inferencer_config']._options = None
   _GIGLRESOURCECONFIG.fields_by_name['inferencer_config']._serialized_options = b'\030\001'
-  _COMPONENT._serialized_start=3166
-  _COMPONENT._serialized_end=3409
+  _COMPONENT._serialized_start=3196
+  _COMPONENT._serialized_end=3439
   _SPARKRESOURCECONFIG._serialized_start=77
   _SPARKRESOURCECONFIG._serialized_end=166
   _DATAFLOWRESOURCECONFIG._serialized_start=168
@@ -182,24 +182,24 @@
   _KFPTRAINERCONFIG._serialized_end=703
   _LOCALTRAINERCONFIG._serialized_start=705
   _LOCALTRAINERCONFIG._serialized_end=746
-  _VERTEXAIRESOURCECONFIG._serialized_start=748
-  _VERTEXAIRESOURCECONFIG._serialized_end=870
-  _KFPRESOURCECONFIG._serialized_start=872
-  _KFPRESOURCECONFIG._serialized_end=995
-  _LOCALRESOURCECONFIG._serialized_start=997
-  _LOCALRESOURCECONFIG._serialized_end=1039
-  _DISTRIBUTEDTRAINERCONFIG._serialized_start=1042
-  _DISTRIBUTEDTRAINERCONFIG._serialized_end=1317
-  _TRAINERRESOURCECONFIG._serialized_start=1320
-  _TRAINERRESOURCECONFIG._serialized_end=1595
-  _INFERENCERRESOURCECONFIG._serialized_start=1598
-  _INFERENCERRESOURCECONFIG._serialized_end=1898
-  _SHAREDRESOURCECONFIG._serialized_start=1901
-  _SHAREDRESOURCECONFIG._serialized_end=2448
-  _SHAREDRESOURCECONFIG_COMMONCOMPUTECONFIG._serialized_start=2114
-  _SHAREDRESOURCECONFIG_COMMONCOMPUTECONFIG._serialized_end=2393
-  _SHAREDRESOURCECONFIG_RESOURCELABELSENTRY._serialized_start=2395
-  _SHAREDRESOURCECONFIG_RESOURCELABELSENTRY._serialized_end=2448
-  _GIGLRESOURCECONFIG._serialized_start=2451
-  _GIGLRESOURCECONFIG._serialized_end=3163
+  _VERTEXAIRESOURCECONFIG._serialized_start=749
+  _VERTEXAIRESOURCECONFIG._serialized_end=900
+  _KFPRESOURCECONFIG._serialized_start=902
+  _KFPRESOURCECONFIG._serialized_end=1025
+  _LOCALRESOURCECONFIG._serialized_start=1027
+  _LOCALRESOURCECONFIG._serialized_end=1069
+  _DISTRIBUTEDTRAINERCONFIG._serialized_start=1072
+  _DISTRIBUTEDTRAINERCONFIG._serialized_end=1347
+  _TRAINERRESOURCECONFIG._serialized_start=1350
+  _TRAINERRESOURCECONFIG._serialized_end=1625
+  _INFERENCERRESOURCECONFIG._serialized_start=1628
+  _INFERENCERRESOURCECONFIG._serialized_end=1928
+  _SHAREDRESOURCECONFIG._serialized_start=1931
+  _SHAREDRESOURCECONFIG._serialized_end=2478
+  _SHAREDRESOURCECONFIG_COMMONCOMPUTECONFIG._serialized_start=2144
+  _SHAREDRESOURCECONFIG_COMMONCOMPUTECONFIG._serialized_end=2423
+  _SHAREDRESOURCECONFIG_RESOURCELABELSENTRY._serialized_start=2425
+  _SHAREDRESOURCECONFIG_RESOURCELABELSENTRY._serialized_end=2478
+  _GIGLRESOURCECONFIG._serialized_start=2481
+  _GIGLRESOURCECONFIG._serialized_end=3193
 # @@protoc_insertion_point(module_scope)
diff --git a/python/snapchat/research/gbml/gigl_resource_config_pb2.pyi b/python/snapchat/research/gbml/gigl_resource_config_pb2.pyi
index 87063f2f5..00ae6817f 100644
--- a/python/snapchat/research/gbml/gigl_resource_config_pb2.pyi
+++ b/python/snapchat/research/gbml/gigl_resource_config_pb2.pyi
@@ -216,6 +216,7 @@ class VertexAiResourceConfig(google.protobuf.message.Message):
     GPU_LIMIT_FIELD_NUMBER: builtins.int
     NUM_REPLICAS_FIELD_NUMBER: builtins.int
     TIMEOUT_FIELD_NUMBER: builtins.int
+    GCP_REGION_OVERRIDE_FIELD_NUMBER: builtins.int
     machine_type: builtins.str
     """Machine type for job"""
     gpu_type: builtins.str
@@ -225,9 +226,15 @@ class VertexAiResourceConfig(google.protobuf.message.Message):
     num_replicas: builtins.int
     """Num workers for job"""
     timeout: builtins.int
-    """Timeout in seconds for the job. If unset or zero, will use the default @ google.cloud.aiplatform.CustomJob, which is 7 days: 
+    """Timeout in seconds for the job. If unset or zero, will use the default @ google.cloud.aiplatform.CustomJob, which is 7 days:
     https://github.com/googleapis/python-aiplatform/blob/58fbabdeeefd1ccf1a9d0c22eeb5606aeb9c2266/google/cloud/aiplatform/jobs.py#L2252-L2253
     """
+    gcp_region_override: builtins.str
+    """Region override.
+    If provided, then the Vertex AI Job will be launched in the provided region.
+    ex: "us-west1"
+    NOTE: If set, then there may be data egress costs from CommonComputeConfig.region -> gcp_region_override
+    """
     def __init__(
         self,
         *,
@@ -236,8 +243,9 @@ class VertexAiResourceConfig(google.protobuf.message.Message):
         gpu_limit: builtins.int = ...,
         num_replicas: builtins.int = ...,
         timeout: builtins.int = ...,
+        gcp_region_override: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["gpu_limit", b"gpu_limit", "gpu_type", b"gpu_type", "machine_type", b"machine_type", "num_replicas", b"num_replicas", "timeout", b"timeout"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["gcp_region_override", b"gcp_region_override", "gpu_limit", b"gpu_limit", "gpu_type", b"gpu_type", "machine_type", b"machine_type", "num_replicas", b"num_replicas", "timeout", b"timeout"]) -> None: ...
 
 global___VertexAiResourceConfig = VertexAiResourceConfig
 
diff --git a/python/tests/unit/src/common/types/pb_wrappers/gigl_resource_config_test.py b/python/tests/unit/src/common/types/pb_wrappers/gigl_resource_config_test.py
new file mode 100644
index 000000000..787aa1f01
--- /dev/null
+++ b/python/tests/unit/src/common/types/pb_wrappers/gigl_resource_config_test.py
@@ -0,0 +1,118 @@
+import unittest
+
+from gigl.src.common.types.pb_wrappers.gigl_resource_config import (
+    GiglResourceConfigWrapper,
+)
+from snapchat.research.gbml.gigl_resource_config_pb2 import GiglResourceConfig
+
+
+class GiglResourceConfigTest(unittest.TestCase):
+    def test_vertex_ai_trainer_region_default(self):
+        resource_config = GiglResourceConfig()
+        resource_config.shared_resource_config.common_compute_config.region = (
+            "us-central1"
+        )
+        resource_config.trainer_resource_config.vertex_ai_trainer_config.SetInParent()
+        resource_config_wrapper = GiglResourceConfigWrapper(resource_config)
+        self.assertEqual(
+            resource_config_wrapper.vertex_ai_trainer_region, "us-central1"
+        )
+
+    def test_vertex_ai_trainer_region_override(self):
+        resource_config = GiglResourceConfig()
+        resource_config.shared_resource_config.common_compute_config.region = (
+            "us-central1"
+        )
+        resource_config.trainer_resource_config.vertex_ai_trainer_config.gcp_region_override = (
+            "us-east1"
+        )
+        resource_config_wrapper = GiglResourceConfigWrapper(resource_config)
+        self.assertEqual(resource_config_wrapper.vertex_ai_trainer_region, "us-east1")
+
+    def test_vertex_ai_inferencer_region_default(self):
+        resource_config = GiglResourceConfig()
+        resource_config.shared_resource_config.common_compute_config.region = (
+            "us-central1"
+        )
+        resource_config.inferencer_resource_config.vertex_ai_inferencer_config.SetInParent()
+        resource_config_wrapper = GiglResourceConfigWrapper(resource_config)
+        self.assertEqual(
+            resource_config_wrapper.vertex_ai_inferencer_region, "us-central1"
+        )
+
+    def test_vertex_ai_inferencer_region_override(self):
+        resource_config = GiglResourceConfig()
+        resource_config.shared_resource_config.common_compute_config.region = (
+            "us-central1"
+        )
+        resource_config.inferencer_resource_config.vertex_ai_inferencer_config.gcp_region_override = (
+            "us-east1"
+        )
+        resource_config_wrapper = GiglResourceConfigWrapper(resource_config)
+        self.assertEqual(
+            resource_config_wrapper.vertex_ai_inferencer_region, "us-east1"
+        )
+
+    def test_vertex_ai_trainer_region_error_no_trainer_config(self):
+        resource_config = GiglResourceConfig()
+        resource_config.shared_resource_config.common_compute_config.region = (
+            "us-central1"
+        )
+        resource_config.trainer_resource_config.SetInParent()
+        resource_config_wrapper = GiglResourceConfigWrapper(resource_config)
+        with self.assertRaises(ValueError):
+            resource_config_wrapper.vertex_ai_trainer_region
+
+    def test_vertex_ai_trainer_region_error_local_trainer_config(self):
+        resource_config = GiglResourceConfig()
+        resource_config.shared_resource_config.common_compute_config.region = (
+            "us-central1"
+        )
+        resource_config.trainer_resource_config.local_trainer_config.SetInParent()
+        resource_config_wrapper = GiglResourceConfigWrapper(resource_config)
+        with self.assertRaises(ValueError):
+            resource_config_wrapper.vertex_ai_trainer_region
+
+    def test_vertex_ai_trainer_region_error_kfp_trainer_config(self):
+        resource_config = GiglResourceConfig()
+        resource_config.shared_resource_config.common_compute_config.region = (
+            "us-central1"
+        )
+        resource_config.trainer_resource_config.kfp_trainer_config.SetInParent()
+        resource_config_wrapper = GiglResourceConfigWrapper(resource_config)
+        with self.assertRaises(ValueError):
+            resource_config_wrapper.vertex_ai_trainer_region
+
+    def test_vertex_ai_inferencer_region_error_no_inferencer_config(self):
+        resource_config = GiglResourceConfig()
+        resource_config.shared_resource_config.common_compute_config.region = (
+            "us-central1"
+        )
+        resource_config.inferencer_resource_config.SetInParent()
+        resource_config_wrapper = GiglResourceConfigWrapper(resource_config)
+        with self.assertRaises(ValueError):
+            resource_config_wrapper.vertex_ai_inferencer_region
+
+    def test_vertex_ai_inferencer_region_error_local_inferencer_config(self):
+        resource_config = GiglResourceConfig()
+        resource_config.shared_resource_config.common_compute_config.region = (
+            "us-central1"
+        )
+        resource_config.inferencer_resource_config.local_inferencer_config.SetInParent()
+        resource_config_wrapper = GiglResourceConfigWrapper(resource_config)
+        with self.assertRaises(ValueError):
+            resource_config_wrapper.vertex_ai_inferencer_region
+
+    def test_vertex_ai_inferencer_region_error_dataflow_inferencer_config(self):
+        resource_config = GiglResourceConfig()
+        resource_config.shared_resource_config.common_compute_config.region = (
+            "us-central1"
+        )
+        resource_config.inferencer_resource_config.dataflow_inferencer_config.SetInParent()
+        resource_config_wrapper = GiglResourceConfigWrapper(resource_config)
+        with self.assertRaises(ValueError):
+            resource_config_wrapper.vertex_ai_inferencer_region
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/scala/common/src/main/scala/snapchat/research/gbml/gbml_config/GbmlConfig.scala b/scala/common/src/main/scala/snapchat/research/gbml/gbml_config/GbmlConfig.scala
index 12adb5a17..b3f110643 100644
--- a/scala/common/src/main/scala/snapchat/research/gbml/gbml_config/GbmlConfig.scala
+++ b/scala/common/src/main/scala/snapchat/research/gbml/gbml_config/GbmlConfig.scala
@@ -73,7 +73,7 @@ final case class GbmlConfig(
         __serializedSizeMemoized = __size
       }
       __size - 1
-      
+
     }
     def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
       taskMetadata.foreach { __v =>
@@ -359,7 +359,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         taskMetadata.nodeBasedTaskMetadata.foreach { __v =>
@@ -411,7 +411,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.type = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TaskMetadata])
   }
-  
+
   object TaskMetadata extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata = {
@@ -491,7 +491,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         override def number: _root_.scala.Int = 0
         override def value: _root_.scala.Nothing = throw new java.util.NoSuchElementException("Empty.value")
       }
-    
+
       @SerialVersionUID(0L)
       final case class NodeBasedTaskMetadata(value: snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata) extends snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.TaskMetadata {
         type ValueType = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata
@@ -537,7 +537,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           supervisionNodeTypes.foreach { __v =>
@@ -567,7 +567,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata.type = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata])
     }
-    
+
     object NodeBasedTaskMetadata extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata = {
@@ -619,7 +619,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata])
     }
-    
+
     @SerialVersionUID(0L)
     final case class NodeAnchorBasedLinkPredictionTaskMetadata(
         supervisionEdgeTypes: _root_.scala.Seq[snapchat.research.gbml.graph_schema.EdgeType] = _root_.scala.Seq.empty,
@@ -643,7 +643,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           supervisionEdgeTypes.foreach { __v =>
@@ -675,7 +675,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata.type = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata])
     }
-    
+
     object NodeAnchorBasedLinkPredictionTaskMetadata extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata = {
@@ -733,7 +733,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata])
     }
-    
+
     @SerialVersionUID(0L)
     final case class LinkBasedTaskMetadata(
         supervisionEdgeTypes: _root_.scala.Seq[snapchat.research.gbml.graph_schema.EdgeType] = _root_.scala.Seq.empty,
@@ -757,7 +757,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           supervisionEdgeTypes.foreach { __v =>
@@ -789,7 +789,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata.type = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata])
     }
-    
+
     object LinkBasedTaskMetadata extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata = {
@@ -847,7 +847,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata])
     }
-    
+
     implicit class TaskMetadataLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata](_l) {
       def nodeBasedTaskMetadata: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata] = field(_.getNodeBasedTaskMetadata)((c_, f_) => c_.copy(taskMetadata = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.TaskMetadata.NodeBasedTaskMetadata(f_)))
       def nodeAnchorBasedLinkPredictionTaskMetadata: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata] = field(_.getNodeAnchorBasedLinkPredictionTaskMetadata)((c_, f_) => c_.copy(taskMetadata = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata(f_)))
@@ -864,7 +864,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TaskMetadata])
   }
-  
+
   /** @param preprocessedMetadataUri
     *   Uri where DataPreprocessor generates the PreprocessedMetadata proto.
     * @param flattenedGraphMetadata
@@ -917,7 +917,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = preprocessedMetadataUri
           if (!__value.isEmpty) {
@@ -948,42 +948,42 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           val __value = snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig._typemapper_sharedArgs.toBase(__item)
           __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
         }
-        
+
         {
           val __value = isGraphDirected
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(7, __value)
           }
         };
-        
+
         {
           val __value = shouldSkipTraining
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(8, __value)
           }
         };
-        
+
         {
           val __value = shouldSkipAutomaticTempAssetCleanup
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(9, __value)
           }
         };
-        
+
         {
           val __value = shouldSkipInference
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(10, __value)
           }
         };
-        
+
         {
           val __value = shouldSkipModelEvaluation
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(11, __value)
           }
         };
-        
+
         {
           val __value = shouldIncludeIsolatedNodesInTraining
           if (__value != false) {
@@ -1000,7 +1000,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -1171,7 +1171,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.SharedConfig])
   }
-  
+
   object SharedConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig = {
@@ -1308,14 +1308,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -1332,7 +1332,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -1376,7 +1376,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.SharedArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.SharedArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.SharedConfig.SharedArgsEntry])
     }
-    
+
     object SharedArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.SharedArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.SharedArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.SharedArgsEntry = {
@@ -1441,7 +1441,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.SharedConfig.SharedArgsEntry])
     }
-    
+
     implicit class SharedConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig](_l) {
       def preprocessedMetadataUri: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.preprocessedMetadataUri)((c_, f_) => c_.copy(preprocessedMetadataUri = f_))
       def flattenedGraphMetadata: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.flattened_graph_metadata.FlattenedGraphMetadata] = field(_.getFlattenedGraphMetadata)((c_, f_) => c_.copy(flattenedGraphMetadata = Option(f_)))
@@ -1508,7 +1508,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.SharedConfig])
   }
-  
+
   /** Contains config related to generating training data for a GML task.
     */
   @SerialVersionUID(0L)
@@ -1544,7 +1544,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         dataPreprocessorConfig.foreach { __v =>
@@ -1597,7 +1597,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig])
   }
-  
+
   object DatasetConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig = {
@@ -1678,7 +1678,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = dataPreprocessorConfigClsPath
             if (!__value.isEmpty) {
@@ -1699,7 +1699,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -1743,7 +1743,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.DataPreprocessorConfig])
     }
-    
+
     object DataPreprocessorConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig = {
@@ -1809,14 +1809,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
           private[this] def __computeSerializedSize(): _root_.scala.Int = {
             var __size = 0
-            
+
             {
               val __value = key
               if (!__value.isEmpty) {
                 __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
               }
             };
-            
+
             {
               val __value = value
               if (!__value.isEmpty) {
@@ -1833,7 +1833,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
               __serializedSizeMemoized = __size
             }
             __size - 1
-            
+
           }
           def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
             {
@@ -1877,7 +1877,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry
           // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry])
       }
-      
+
       object DataPreprocessorArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry] {
         implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry] = this
         def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry = {
@@ -1942,7 +1942,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         )
         // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry])
       }
-      
+
       implicit class DataPreprocessorConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig](_l) {
         def dataPreprocessorConfigClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.dataPreprocessorConfigClsPath)((c_, f_) => c_.copy(dataPreprocessorConfigClsPath = f_))
         def dataPreprocessorArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.dataPreprocessorArgs)((c_, f_) => c_.copy(dataPreprocessorArgs = f_))
@@ -1960,7 +1960,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.DataPreprocessorConfig])
     }
-    
+
     /** @param numHops
       *   number of hops for subgraph sampler to include
       * @param numNeighborsToSample
@@ -1990,7 +1990,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       *   Also used in loss computation. Random negatives will always be used even when there
       *   are no user defined hard negatives
       * @param graphDbConfig
-      *   If specified, intention is to run ingestion into graphDB for subgraph sampler 
+      *   If specified, intention is to run ingestion into graphDB for subgraph sampler
       */
     @SerialVersionUID(0L)
     final case class SubgraphSamplerConfig(
@@ -2009,14 +2009,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = numHops
             if (__value != 0) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(1, __value)
             }
           };
-          
+
           {
             val __value = numNeighborsToSample
             if (__value != 0) {
@@ -2027,7 +2027,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             val __value = subgraphSamplingStrategy.get
             __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
           };
-          
+
           {
             val __value = numPositiveSamples
             if (__value != 0) {
@@ -2038,21 +2038,21 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             val __value = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig._typemapper_experimentalFlags.toBase(__item)
             __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
           }
-          
+
           {
             val __value = numMaxTrainingSamplesToOutput
             if (__value != 0) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(6, __value)
             }
           };
-          
+
           {
             val __value = numUserDefinedPositiveSamples
             if (__value != 0) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(7, __value)
             }
           };
-          
+
           {
             val __value = numUserDefinedNegativeSamples
             if (__value != 0) {
@@ -2073,7 +2073,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -2199,7 +2199,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.SubgraphSamplerConfig])
     }
-    
+
     object SubgraphSamplerConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig = {
@@ -2309,14 +2309,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
           private[this] def __computeSerializedSize(): _root_.scala.Int = {
             var __size = 0
-            
+
             {
               val __value = key
               if (!__value.isEmpty) {
                 __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
               }
             };
-            
+
             {
               val __value = value
               if (!__value.isEmpty) {
@@ -2333,7 +2333,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
               __serializedSizeMemoized = __size
             }
             __size - 1
-            
+
           }
           def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
             {
@@ -2377,7 +2377,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry
           // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry])
       }
-      
+
       object ExperimentalFlagsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry] {
         implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry] = this
         def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry = {
@@ -2442,7 +2442,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         )
         // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry])
       }
-      
+
       implicit class SubgraphSamplerConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig](_l) {
         def numHops: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Int] = field(_.numHops)((c_, f_) => c_.copy(numHops = f_))
         def numNeighborsToSample: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Int] = field(_.numNeighborsToSample)((c_, f_) => c_.copy(numNeighborsToSample = f_))
@@ -2490,7 +2490,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.SubgraphSamplerConfig])
     }
-    
+
     /** @param splitStrategyClsPath
       *   Module path to concrete SplitStrategy instance.
       * @param splitStrategyArgs
@@ -2512,7 +2512,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = splitStrategyClsPath
             if (!__value.isEmpty) {
@@ -2523,7 +2523,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             val __value = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig._typemapper_splitStrategyArgs.toBase(__item)
             __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
           }
-          
+
           {
             val __value = assignerClsPath
             if (!__value.isEmpty) {
@@ -2544,7 +2544,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -2612,7 +2612,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig])
     }
-    
+
     object SplitGeneratorConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig = {
@@ -2692,14 +2692,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
           private[this] def __computeSerializedSize(): _root_.scala.Int = {
             var __size = 0
-            
+
             {
               val __value = key
               if (!__value.isEmpty) {
                 __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
               }
             };
-            
+
             {
               val __value = value
               if (!__value.isEmpty) {
@@ -2716,7 +2716,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
               __serializedSizeMemoized = __size
             }
             __size - 1
-            
+
           }
           def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
             {
@@ -2760,7 +2760,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry
           // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry])
       }
-      
+
       object SplitStrategyArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry] {
         implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry] = this
         def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry = {
@@ -2825,7 +2825,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         )
         // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry])
       }
-      
+
       @SerialVersionUID(0L)
       final case class AssignerArgsEntry(
           key: _root_.scala.Predef.String = "",
@@ -2836,14 +2836,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
           private[this] def __computeSerializedSize(): _root_.scala.Int = {
             var __size = 0
-            
+
             {
               val __value = key
               if (!__value.isEmpty) {
                 __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
               }
             };
-            
+
             {
               val __value = value
               if (!__value.isEmpty) {
@@ -2860,7 +2860,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
               __serializedSizeMemoized = __size
             }
             __size - 1
-            
+
           }
           def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
             {
@@ -2904,7 +2904,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry
           // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry])
       }
-      
+
       object AssignerArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry] {
         implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry] = this
         def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry = {
@@ -2969,7 +2969,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         )
         // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry])
       }
-      
+
       implicit class SplitGeneratorConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig](_l) {
         def splitStrategyClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.splitStrategyClsPath)((c_, f_) => c_.copy(splitStrategyClsPath = f_))
         def splitStrategyArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.splitStrategyArgs)((c_, f_) => c_.copy(splitStrategyArgs = f_))
@@ -2997,7 +2997,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig])
     }
-    
+
     implicit class DatasetConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig](_l) {
       def dataPreprocessorConfig: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig] = field(_.getDataPreprocessorConfig)((c_, f_) => c_.copy(dataPreprocessorConfig = Option(f_)))
       def optionalDataPreprocessorConfig: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Option[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig]] = field(_.dataPreprocessorConfig)((c_, f_) => c_.copy(dataPreprocessorConfig = f_))
@@ -3020,11 +3020,11 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig])
   }
-  
+
   /** Generic Configuration for a GraphDB connection.
     *
     * @param graphDbIngestionClsPath
-    *   Python class path pointing to user-written 
+    *   Python class path pointing to user-written
     *   `BaseIngestion`` class definition. e.g. `my.team.graph_db.BaseInjectionImpl`.
     *   This class is currently, as an implementation detail, used for injestion only.
     *   We document this *purely* for information purposes and may change the implementation at any time.
@@ -3052,7 +3052,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = graphDbIngestionClsPath
           if (!__value.isEmpty) {
@@ -3081,7 +3081,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -3148,7 +3148,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.GraphDBConfig])
   }
-  
+
   object GraphDBConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig = {
@@ -3230,14 +3230,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -3254,7 +3254,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -3298,7 +3298,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry])
     }
-    
+
     object GraphDbIngestionArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry = {
@@ -3363,7 +3363,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry])
     }
-    
+
     @SerialVersionUID(0L)
     final case class GraphDbArgsEntry(
         key: _root_.scala.Predef.String = "",
@@ -3374,14 +3374,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -3398,7 +3398,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -3442,7 +3442,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDbArgsEntry])
     }
-    
+
     object GraphDbArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbArgsEntry = {
@@ -3507,7 +3507,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDbArgsEntry])
     }
-    
+
     /** Scala-specific configuration.
       *
       * @param graphDbClientClassPath
@@ -3523,7 +3523,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = graphDbClientClassPath
             if (!__value.isEmpty) {
@@ -3540,7 +3540,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -3572,7 +3572,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDBServiceConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDBServiceConfig
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDBServiceConfig])
     }
-    
+
     object GraphDBServiceConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDBServiceConfig] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDBServiceConfig] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDBServiceConfig = {
@@ -3624,7 +3624,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDBServiceConfig])
     }
-    
+
     implicit class GraphDBConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig](_l) {
       def graphDbIngestionClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.graphDbIngestionClsPath)((c_, f_) => c_.copy(graphDbIngestionClsPath = f_))
       def graphDbIngestionArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.graphDbIngestionArgs)((c_, f_) => c_.copy(graphDbIngestionArgs = f_))
@@ -3653,10 +3653,10 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.GraphDBConfig])
   }
-  
+
   /** @param trainerClsPath
     *   (deprecated)
-    *   Uri pointing to user-written BaseTrainer class definition. Used for the subgraph-sampling-based training process. 
+    *   Uri pointing to user-written BaseTrainer class definition. Used for the subgraph-sampling-based training process.
     * @param trainerArgs
     *   Arguments to parameterize training process with.
     * @param shouldLogToTensorboard
@@ -3674,7 +3674,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = trainerClsPath
           if (!__value.isEmpty) {
@@ -3693,7 +3693,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           val __value = executable.command.get
           __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(101, __value)
         };
-        
+
         {
           val __value = shouldLogToTensorboard
           if (__value != false) {
@@ -3710,7 +3710,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -3784,7 +3784,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TrainerConfig])
   }
-  
+
   object TrainerConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig = {
@@ -3873,7 +3873,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         override def number: _root_.scala.Int = 0
         override def value: _root_.scala.Nothing = throw new java.util.NoSuchElementException("Empty.value")
       }
-    
+
       @SerialVersionUID(0L)
       final case class ClsPath(value: _root_.scala.Predef.String) extends snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.Executable {
         type ValueType = _root_.scala.Predef.String
@@ -3899,14 +3899,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -3923,7 +3923,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -3967,7 +3967,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.TrainerArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.TrainerArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TrainerConfig.TrainerArgsEntry])
     }
-    
+
     object TrainerArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.TrainerArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.TrainerArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.TrainerArgsEntry = {
@@ -4032,7 +4032,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TrainerConfig.TrainerArgsEntry])
     }
-    
+
     implicit class TrainerConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig](_l) {
       def trainerClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.trainerClsPath)((c_, f_) => c_.copy(trainerClsPath = f_))
       def trainerArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.trainerArgs)((c_, f_) => c_.copy(trainerArgs = f_))
@@ -4061,10 +4061,10 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TrainerConfig])
   }
-  
+
   /** @param inferencerClsPath
     *   (deprecated)
-    *   Path to modeling task spec class path to construct model for inference. Used for the subgraph-sampling-based inference process. 
+    *   Path to modeling task spec class path to construct model for inference. Used for the subgraph-sampling-based inference process.
     * @param inferenceBatchSize
     *   Optional. If set, will be used to batch inference samples to a specific size before call for inference is made
     *   Defaults to setting in python/gigl/src/inference/gnn_inferencer.py
@@ -4085,7 +4085,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           val __value = snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig._typemapper_inferencerArgs.toBase(__item)
           __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
         }
-        
+
         {
           val __value = inferencerClsPath
           if (!__value.isEmpty) {
@@ -4100,7 +4100,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           val __value = executable.command.get
           __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(101, __value)
         };
-        
+
         {
           val __value = inferenceBatchSize
           if (__value != 0) {
@@ -4117,7 +4117,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         inferencerArgs.foreach { __v =>
@@ -4191,7 +4191,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.InferencerConfig])
   }
-  
+
   object InferencerConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig = {
@@ -4280,7 +4280,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         override def number: _root_.scala.Int = 0
         override def value: _root_.scala.Nothing = throw new java.util.NoSuchElementException("Empty.value")
       }
-    
+
       @SerialVersionUID(0L)
       final case class ClsPath(value: _root_.scala.Predef.String) extends snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.Executable {
         type ValueType = _root_.scala.Predef.String
@@ -4306,14 +4306,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -4330,7 +4330,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -4374,7 +4374,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.InferencerArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.InferencerArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.InferencerConfig.InferencerArgsEntry])
     }
-    
+
     object InferencerArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.InferencerArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.InferencerArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.InferencerArgsEntry = {
@@ -4439,7 +4439,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.InferencerConfig.InferencerArgsEntry])
     }
-    
+
     implicit class InferencerConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig](_l) {
       def inferencerArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.inferencerArgs)((c_, f_) => c_.copy(inferencerArgs = f_))
       def inferencerClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.inferencerClsPath)((c_, f_) => c_.copy(inferencerClsPath = f_))
@@ -4468,7 +4468,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.InferencerConfig])
   }
-  
+
   @SerialVersionUID(0L)
   final case class PostProcessorConfig(
       postProcessorArgs: _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String] = _root_.scala.collection.immutable.Map.empty,
@@ -4483,7 +4483,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           val __value = snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig._typemapper_postProcessorArgs.toBase(__item)
           __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
         }
-        
+
         {
           val __value = postProcessorClsPath
           if (!__value.isEmpty) {
@@ -4500,7 +4500,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         postProcessorArgs.foreach { __v =>
@@ -4544,7 +4544,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.PostProcessorConfig])
   }
-  
+
   object PostProcessorConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig = {
@@ -4610,14 +4610,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -4634,7 +4634,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -4678,7 +4678,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry])
     }
-    
+
     object PostProcessorArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry = {
@@ -4743,7 +4743,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry])
     }
-    
+
     implicit class PostProcessorConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig](_l) {
       def postProcessorArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.postProcessorArgs)((c_, f_) => c_.copy(postProcessorArgs = f_))
       def postProcessorClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.postProcessorClsPath)((c_, f_) => c_.copy(postProcessorClsPath = f_))
@@ -4761,7 +4761,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.PostProcessorConfig])
   }
-  
+
   @SerialVersionUID(0L)
   final case class MetricsConfig(
       metricsClsPath: _root_.scala.Predef.String = "",
@@ -4772,7 +4772,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = metricsClsPath
           if (!__value.isEmpty) {
@@ -4793,7 +4793,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -4837,7 +4837,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.MetricsConfig])
   }
-  
+
   object MetricsConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig = {
@@ -4903,14 +4903,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -4927,7 +4927,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -4971,7 +4971,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.MetricsArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.MetricsArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.MetricsConfig.MetricsArgsEntry])
     }
-    
+
     object MetricsArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.MetricsArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.MetricsArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.MetricsArgsEntry = {
@@ -5036,7 +5036,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.MetricsConfig.MetricsArgsEntry])
     }
-    
+
     implicit class MetricsConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig](_l) {
       def metricsClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.metricsClsPath)((c_, f_) => c_.copy(metricsClsPath = f_))
       def metricsArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.metricsArgs)((c_, f_) => c_.copy(metricsArgs = f_))
@@ -5054,7 +5054,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.MetricsConfig])
   }
-  
+
   @SerialVersionUID(0L)
   final case class ProfilerConfig(
       shouldEnableProfiler: _root_.scala.Boolean = false,
@@ -5066,14 +5066,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = shouldEnableProfiler
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(1, __value)
           }
         };
-        
+
         {
           val __value = profilerLogDir
           if (!__value.isEmpty) {
@@ -5094,7 +5094,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -5150,7 +5150,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.ProfilerConfig])
   }
-  
+
   object ProfilerConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig = {
@@ -5222,14 +5222,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -5246,7 +5246,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -5290,7 +5290,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.ProfilerArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.ProfilerArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.ProfilerConfig.ProfilerArgsEntry])
     }
-    
+
     object ProfilerArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.ProfilerArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.ProfilerArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.ProfilerArgsEntry = {
@@ -5355,7 +5355,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.ProfilerConfig.ProfilerArgsEntry])
     }
-    
+
     implicit class ProfilerConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig](_l) {
       def shouldEnableProfiler: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Boolean] = field(_.shouldEnableProfiler)((c_, f_) => c_.copy(shouldEnableProfiler = f_))
       def profilerLogDir: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.profilerLogDir)((c_, f_) => c_.copy(profilerLogDir = f_))
@@ -5377,7 +5377,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.ProfilerConfig])
   }
-  
+
   @SerialVersionUID(0L)
   final case class FeatureFlagsEntry(
       key: _root_.scala.Predef.String = "",
@@ -5388,14 +5388,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = key
           if (!__value.isEmpty) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
           }
         };
-        
+
         {
           val __value = value
           if (!__value.isEmpty) {
@@ -5412,7 +5412,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -5456,7 +5456,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.FeatureFlagsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.FeatureFlagsEntry
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.FeatureFlagsEntry])
   }
-  
+
   object FeatureFlagsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.FeatureFlagsEntry] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.FeatureFlagsEntry] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.FeatureFlagsEntry = {
@@ -5521,7 +5521,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.FeatureFlagsEntry])
   }
-  
+
   implicit class GbmlConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig](_l) {
     def taskMetadata: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata] = field(_.getTaskMetadata)((c_, f_) => c_.copy(taskMetadata = Option(f_)))
     def optionalTaskMetadata: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Option[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata]] = field(_.taskMetadata)((c_, f_) => c_.copy(taskMetadata = f_))
diff --git a/scala/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/GiglResourceConfigProto.scala b/scala/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/GiglResourceConfigProto.scala
index c03fc503b..55b0dfcf7 100644
--- a/scala/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/GiglResourceConfigProto.scala
+++ b/scala/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/GiglResourceConfigProto.scala
@@ -44,68 +44,69 @@ object GiglResourceConfigProto extends _root_.scalapb.GeneratedFileObject {
   W1vcnlSZXF1ZXN0Ug1tZW1vcnlSZXF1ZXN0EicKCGdwdV90eXBlGAMgASgJQgziPwkSB2dwdVR5cGVSB2dwdVR5cGUSKgoJZ3B1X
   2xpbWl0GAQgASgNQg3iPwoSCGdwdUxpbWl0UghncHVMaW1pdBIzCgxudW1fcmVwbGljYXMYBSABKA1CEOI/DRILbnVtUmVwbGljY
   XNSC251bVJlcGxpY2FzIkYKEkxvY2FsVHJhaW5lckNvbmZpZxIwCgtudW1fd29ya2VycxgBIAEoDUIP4j8MEgpudW1Xb3JrZXJzU
-  gpudW1Xb3JrZXJzIv8BChZWZXJ0ZXhBaVJlc291cmNlQ29uZmlnEjMKDG1hY2hpbmVfdHlwZRgBIAEoCUIQ4j8NEgttYWNoaW5lV
+  gpudW1Xb3JrZXJzIscCChZWZXJ0ZXhBaVJlc291cmNlQ29uZmlnEjMKDG1hY2hpbmVfdHlwZRgBIAEoCUIQ4j8NEgttYWNoaW5lV
   HlwZVILbWFjaGluZVR5cGUSJwoIZ3B1X3R5cGUYAiABKAlCDOI/CRIHZ3B1VHlwZVIHZ3B1VHlwZRIqCglncHVfbGltaXQYAyABK
   A1CDeI/ChIIZ3B1TGltaXRSCGdwdUxpbWl0EjMKDG51bV9yZXBsaWNhcxgEIAEoDUIQ4j8NEgtudW1SZXBsaWNhc1ILbnVtUmVwb
-  GljYXMSJgoHdGltZW91dBgFIAEoDUIM4j8JEgd0aW1lb3V0Ugd0aW1lb3V0IooCChFLRlBSZXNvdXJjZUNvbmZpZxIwCgtjcHVfc
-  mVxdWVzdBgBIAEoCUIP4j8MEgpjcHVSZXF1ZXN0UgpjcHVSZXF1ZXN0EjkKDm1lbW9yeV9yZXF1ZXN0GAIgASgJQhLiPw8SDW1lb
-  W9yeVJlcXVlc3RSDW1lbW9yeVJlcXVlc3QSJwoIZ3B1X3R5cGUYAyABKAlCDOI/CRIHZ3B1VHlwZVIHZ3B1VHlwZRIqCglncHVfb
-  GltaXQYBCABKA1CDeI/ChIIZ3B1TGltaXRSCGdwdUxpbWl0EjMKDG51bV9yZXBsaWNhcxgFIAEoDUIQ4j8NEgtudW1SZXBsaWNhc
-  1ILbnVtUmVwbGljYXMiRwoTTG9jYWxSZXNvdXJjZUNvbmZpZxIwCgtudW1fd29ya2VycxgBIAEoDUIP4j8MEgpudW1Xb3JrZXJzU
-  gpudW1Xb3JrZXJzIp0DChhEaXN0cmlidXRlZFRyYWluZXJDb25maWcShAEKGHZlcnRleF9haV90cmFpbmVyX2NvbmZpZxgBIAEoC
-  zItLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuVmVydGV4QWlUcmFpbmVyQ29uZmlnQhriPxcSFXZlcnRleEFpVHJhaW5lckNvbmZpZ
-  0gAUhV2ZXJ0ZXhBaVRyYWluZXJDb25maWcSbwoSa2ZwX3RyYWluZXJfY29uZmlnGAIgASgLMiguc25hcGNoYXQucmVzZWFyY2guZ
-  2JtbC5LRlBUcmFpbmVyQ29uZmlnQhXiPxISEGtmcFRyYWluZXJDb25maWdIAFIQa2ZwVHJhaW5lckNvbmZpZxJ3ChRsb2NhbF90c
-  mFpbmVyX2NvbmZpZxgDIAEoCzIqLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuTG9jYWxUcmFpbmVyQ29uZmlnQhfiPxQSEmxvY2FsV
-  HJhaW5lckNvbmZpZ0gAUhJsb2NhbFRyYWluZXJDb25maWdCEAoOdHJhaW5lcl9jb25maWcinQMKFVRyYWluZXJSZXNvdXJjZUNvb
-  mZpZxKFAQoYdmVydGV4X2FpX3RyYWluZXJfY29uZmlnGAEgASgLMi4uc25hcGNoYXQucmVzZWFyY2guZ2JtbC5WZXJ0ZXhBaVJlc
-  291cmNlQ29uZmlnQhriPxcSFXZlcnRleEFpVHJhaW5lckNvbmZpZ0gAUhV2ZXJ0ZXhBaVRyYWluZXJDb25maWcScAoSa2ZwX3RyY
-  WluZXJfY29uZmlnGAIgASgLMikuc25hcGNoYXQucmVzZWFyY2guZ2JtbC5LRlBSZXNvdXJjZUNvbmZpZ0IV4j8SEhBrZnBUcmFpb
-  mVyQ29uZmlnSABSEGtmcFRyYWluZXJDb25maWcSeAoUbG9jYWxfdHJhaW5lcl9jb25maWcYAyABKAsyKy5zbmFwY2hhdC5yZXNlY
-  XJjaC5nYm1sLkxvY2FsUmVzb3VyY2VDb25maWdCF+I/FBISbG9jYWxUcmFpbmVyQ29uZmlnSABSEmxvY2FsVHJhaW5lckNvbmZpZ
-  0IQCg50cmFpbmVyX2NvbmZpZyLUAwoYSW5mZXJlbmNlclJlc291cmNlQ29uZmlnEo4BCht2ZXJ0ZXhfYWlfaW5mZXJlbmNlcl9jb
-  25maWcYASABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLlZlcnRleEFpUmVzb3VyY2VDb25maWdCHeI/GhIYdmVydGV4QWlJb
-  mZlcmVuY2VyQ29uZmlnSABSGHZlcnRleEFpSW5mZXJlbmNlckNvbmZpZxKNAQoaZGF0YWZsb3dfaW5mZXJlbmNlcl9jb25maWcYA
-  iABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRhdGFmbG93UmVzb3VyY2VDb25maWdCHeI/GhIYZGF0YWZsb3dJbmZlcmVuY
-  2VyQ29uZmlnSABSGGRhdGFmbG93SW5mZXJlbmNlckNvbmZpZxKBAQoXbG9jYWxfaW5mZXJlbmNlcl9jb25maWcYAyABKAsyKy5zb
-  mFwY2hhdC5yZXNlYXJjaC5nYm1sLkxvY2FsUmVzb3VyY2VDb25maWdCGuI/FxIVbG9jYWxJbmZlcmVuY2VyQ29uZmlnSABSFWxvY
-  2FsSW5mZXJlbmNlckNvbmZpZ0ITChFpbmZlcmVuY2VyX2NvbmZpZyKXCAoUU2hhcmVkUmVzb3VyY2VDb25maWcSfgoPcmVzb3VyY
-  2VfbGFiZWxzGAEgAygLMkAuc25hcGNoYXQucmVzZWFyY2guZ2JtbC5TaGFyZWRSZXNvdXJjZUNvbmZpZy5SZXNvdXJjZUxhYmVsc
-  0VudHJ5QhPiPxASDnJlc291cmNlTGFiZWxzUg5yZXNvdXJjZUxhYmVscxKOAQoVY29tbW9uX2NvbXB1dGVfY29uZmlnGAIgASgLM
-  kAuc25hcGNoYXQucmVzZWFyY2guZ2JtbC5TaGFyZWRSZXNvdXJjZUNvbmZpZy5Db21tb25Db21wdXRlQ29uZmlnQhjiPxUSE2Nvb
-  W1vbkNvbXB1dGVDb25maWdSE2NvbW1vbkNvbXB1dGVDb25maWcalAUKE0NvbW1vbkNvbXB1dGVDb25maWcSJgoHcHJvamVjdBgBI
-  AEoCUIM4j8JEgdwcm9qZWN0Ugdwcm9qZWN0EiMKBnJlZ2lvbhgCIAEoCUIL4j8IEgZyZWdpb25SBnJlZ2lvbhJDChJ0ZW1wX2Fzc
-  2V0c19idWNrZXQYAyABKAlCFeI/EhIQdGVtcEFzc2V0c0J1Y2tldFIQdGVtcEFzc2V0c0J1Y2tldBJcCht0ZW1wX3JlZ2lvbmFsX
-  2Fzc2V0c19idWNrZXQYBCABKAlCHeI/GhIYdGVtcFJlZ2lvbmFsQXNzZXRzQnVja2V0Uhh0ZW1wUmVnaW9uYWxBc3NldHNCdWNrZ
-  XQSQwoScGVybV9hc3NldHNfYnVja2V0GAUgASgJQhXiPxISEHBlcm1Bc3NldHNCdWNrZXRSEHBlcm1Bc3NldHNCdWNrZXQSWgobd
-  GVtcF9hc3NldHNfYnFfZGF0YXNldF9uYW1lGAYgASgJQhziPxkSF3RlbXBBc3NldHNCcURhdGFzZXROYW1lUhd0ZW1wQXNzZXRzQ
-  nFEYXRhc2V0TmFtZRJWChllbWJlZGRpbmdfYnFfZGF0YXNldF9uYW1lGAcgASgJQhviPxgSFmVtYmVkZGluZ0JxRGF0YXNldE5hb
-  WVSFmVtYmVkZGluZ0JxRGF0YXNldE5hbWUSVgoZZ2NwX3NlcnZpY2VfYWNjb3VudF9lbWFpbBgIIAEoCUIb4j8YEhZnY3BTZXJ2a
-  WNlQWNjb3VudEVtYWlsUhZnY3BTZXJ2aWNlQWNjb3VudEVtYWlsEjwKD2RhdGFmbG93X3J1bm5lchgLIAEoCUIT4j8QEg5kYXRhZ
-  mxvd1J1bm5lclIOZGF0YWZsb3dSdW5uZXIaVwoTUmVzb3VyY2VMYWJlbHNFbnRyeRIaCgNrZXkYASABKAlCCOI/BRIDa2V5UgNrZ
-  XkSIAoFdmFsdWUYAiABKAlCCuI/BxIFdmFsdWVSBXZhbHVlOgI4ASL3CAoSR2lnbFJlc291cmNlQ29uZmlnElsKGnNoYXJlZF9yZ
-  XNvdXJjZV9jb25maWdfdXJpGAEgASgJQhziPxkSF3NoYXJlZFJlc291cmNlQ29uZmlnVXJpSABSF3NoYXJlZFJlc291cmNlQ29uZ
-  mlnVXJpEn8KFnNoYXJlZF9yZXNvdXJjZV9jb25maWcYAiABKAsyLC5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLlNoYXJlZFJlc291c
-  mNlQ29uZmlnQhniPxYSFHNoYXJlZFJlc291cmNlQ29uZmlnSABSFHNoYXJlZFJlc291cmNlQ29uZmlnEngKE3ByZXByb2Nlc3Nvc
-  l9jb25maWcYDCABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRhdGFQcmVwcm9jZXNzb3JDb25maWdCF+I/FBIScHJlcHJvY
-  2Vzc29yQ29uZmlnUhJwcmVwcm9jZXNzb3JDb25maWcSfwoXc3ViZ3JhcGhfc2FtcGxlcl9jb25maWcYDSABKAsyKy5zbmFwY2hhd
-  C5yZXNlYXJjaC5nYm1sLlNwYXJrUmVzb3VyY2VDb25maWdCGuI/FxIVc3ViZ3JhcGhTYW1wbGVyQ29uZmlnUhVzdWJncmFwaFNhb
-  XBsZXJDb25maWcSfAoWc3BsaXRfZ2VuZXJhdG9yX2NvbmZpZxgOIAEoCzIrLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuU3BhcmtSZ
-  XNvdXJjZUNvbmZpZ0IZ4j8WEhRzcGxpdEdlbmVyYXRvckNvbmZpZ1IUc3BsaXRHZW5lcmF0b3JDb25maWcSbQoOdHJhaW5lcl9jb
-  25maWcYDyABKAsyMC5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRpc3RyaWJ1dGVkVHJhaW5lckNvbmZpZ0IUGAHiPw8SDXRyYWluZ
-  XJDb25maWdSDXRyYWluZXJDb25maWcSdAoRaW5mZXJlbmNlcl9jb25maWcYECABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sL
-  kRhdGFmbG93UmVzb3VyY2VDb25maWdCFxgB4j8SEhBpbmZlcmVuY2VyQ29uZmlnUhBpbmZlcmVuY2VyQ29uZmlnEoEBChd0cmFpb
-  mVyX3Jlc291cmNlX2NvbmZpZxgRIAEoCzItLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuVHJhaW5lclJlc291cmNlQ29uZmlnQhriP
-  xcSFXRyYWluZXJSZXNvdXJjZUNvbmZpZ1IVdHJhaW5lclJlc291cmNlQ29uZmlnEo0BChppbmZlcmVuY2VyX3Jlc291cmNlX2Nvb
-  mZpZxgSIAEoCzIwLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuSW5mZXJlbmNlclJlc291cmNlQ29uZmlnQh3iPxoSGGluZmVyZW5jZ
-  XJSZXNvdXJjZUNvbmZpZ1IYaW5mZXJlbmNlclJlc291cmNlQ29uZmlnQhEKD3NoYXJlZF9yZXNvdXJjZSrjAwoJQ29tcG9uZW50E
-  i0KEUNvbXBvbmVudF9Vbmtub3duEAAaFuI/ExIRQ29tcG9uZW50X1Vua25vd24SPwoaQ29tcG9uZW50X0NvbmZpZ19WYWxpZGF0b
-  3IQARof4j8cEhpDb21wb25lbnRfQ29uZmlnX1ZhbGlkYXRvchI/ChpDb21wb25lbnRfQ29uZmlnX1BvcHVsYXRvchACGh/iPxwSG
-  kNvbXBvbmVudF9Db25maWdfUG9wdWxhdG9yEkEKG0NvbXBvbmVudF9EYXRhX1ByZXByb2Nlc3NvchADGiDiPx0SG0NvbXBvbmVud
-  F9EYXRhX1ByZXByb2Nlc3NvchI/ChpDb21wb25lbnRfU3ViZ3JhcGhfU2FtcGxlchAEGh/iPxwSGkNvbXBvbmVudF9TdWJncmFwa
-  F9TYW1wbGVyEj0KGUNvbXBvbmVudF9TcGxpdF9HZW5lcmF0b3IQBRoe4j8bEhlDb21wb25lbnRfU3BsaXRfR2VuZXJhdG9yEi0KE
-  UNvbXBvbmVudF9UcmFpbmVyEAYaFuI/ExIRQ29tcG9uZW50X1RyYWluZXISMwoUQ29tcG9uZW50X0luZmVyZW5jZXIQBxoZ4j8WE
-  hRDb21wb25lbnRfSW5mZXJlbmNlcmIGcHJvdG8z"""
+  GljYXMSJgoHdGltZW91dBgFIAEoDUIM4j8JEgd0aW1lb3V0Ugd0aW1lb3V0EkYKE2djcF9yZWdpb25fb3ZlcnJpZGUYBiABKAlCF
+  uI/ExIRZ2NwUmVnaW9uT3ZlcnJpZGVSEWdjcFJlZ2lvbk92ZXJyaWRlIooCChFLRlBSZXNvdXJjZUNvbmZpZxIwCgtjcHVfcmVxd
+  WVzdBgBIAEoCUIP4j8MEgpjcHVSZXF1ZXN0UgpjcHVSZXF1ZXN0EjkKDm1lbW9yeV9yZXF1ZXN0GAIgASgJQhLiPw8SDW1lbW9ye
+  VJlcXVlc3RSDW1lbW9yeVJlcXVlc3QSJwoIZ3B1X3R5cGUYAyABKAlCDOI/CRIHZ3B1VHlwZVIHZ3B1VHlwZRIqCglncHVfbGlta
+  XQYBCABKA1CDeI/ChIIZ3B1TGltaXRSCGdwdUxpbWl0EjMKDG51bV9yZXBsaWNhcxgFIAEoDUIQ4j8NEgtudW1SZXBsaWNhc1ILb
+  nVtUmVwbGljYXMiRwoTTG9jYWxSZXNvdXJjZUNvbmZpZxIwCgtudW1fd29ya2VycxgBIAEoDUIP4j8MEgpudW1Xb3JrZXJzUgpud
+  W1Xb3JrZXJzIp0DChhEaXN0cmlidXRlZFRyYWluZXJDb25maWcShAEKGHZlcnRleF9haV90cmFpbmVyX2NvbmZpZxgBIAEoCzItL
+  nNuYXBjaGF0LnJlc2VhcmNoLmdibWwuVmVydGV4QWlUcmFpbmVyQ29uZmlnQhriPxcSFXZlcnRleEFpVHJhaW5lckNvbmZpZ0gAU
+  hV2ZXJ0ZXhBaVRyYWluZXJDb25maWcSbwoSa2ZwX3RyYWluZXJfY29uZmlnGAIgASgLMiguc25hcGNoYXQucmVzZWFyY2guZ2Jtb
+  C5LRlBUcmFpbmVyQ29uZmlnQhXiPxISEGtmcFRyYWluZXJDb25maWdIAFIQa2ZwVHJhaW5lckNvbmZpZxJ3ChRsb2NhbF90cmFpb
+  mVyX2NvbmZpZxgDIAEoCzIqLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuTG9jYWxUcmFpbmVyQ29uZmlnQhfiPxQSEmxvY2FsVHJha
+  W5lckNvbmZpZ0gAUhJsb2NhbFRyYWluZXJDb25maWdCEAoOdHJhaW5lcl9jb25maWcinQMKFVRyYWluZXJSZXNvdXJjZUNvbmZpZ
+  xKFAQoYdmVydGV4X2FpX3RyYWluZXJfY29uZmlnGAEgASgLMi4uc25hcGNoYXQucmVzZWFyY2guZ2JtbC5WZXJ0ZXhBaVJlc291c
+  mNlQ29uZmlnQhriPxcSFXZlcnRleEFpVHJhaW5lckNvbmZpZ0gAUhV2ZXJ0ZXhBaVRyYWluZXJDb25maWcScAoSa2ZwX3RyYWluZ
+  XJfY29uZmlnGAIgASgLMikuc25hcGNoYXQucmVzZWFyY2guZ2JtbC5LRlBSZXNvdXJjZUNvbmZpZ0IV4j8SEhBrZnBUcmFpbmVyQ
+  29uZmlnSABSEGtmcFRyYWluZXJDb25maWcSeAoUbG9jYWxfdHJhaW5lcl9jb25maWcYAyABKAsyKy5zbmFwY2hhdC5yZXNlYXJja
+  C5nYm1sLkxvY2FsUmVzb3VyY2VDb25maWdCF+I/FBISbG9jYWxUcmFpbmVyQ29uZmlnSABSEmxvY2FsVHJhaW5lckNvbmZpZ0IQC
+  g50cmFpbmVyX2NvbmZpZyLUAwoYSW5mZXJlbmNlclJlc291cmNlQ29uZmlnEo4BCht2ZXJ0ZXhfYWlfaW5mZXJlbmNlcl9jb25ma
+  WcYASABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLlZlcnRleEFpUmVzb3VyY2VDb25maWdCHeI/GhIYdmVydGV4QWlJbmZlc
+  mVuY2VyQ29uZmlnSABSGHZlcnRleEFpSW5mZXJlbmNlckNvbmZpZxKNAQoaZGF0YWZsb3dfaW5mZXJlbmNlcl9jb25maWcYAiABK
+  AsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRhdGFmbG93UmVzb3VyY2VDb25maWdCHeI/GhIYZGF0YWZsb3dJbmZlcmVuY2VyQ
+  29uZmlnSABSGGRhdGFmbG93SW5mZXJlbmNlckNvbmZpZxKBAQoXbG9jYWxfaW5mZXJlbmNlcl9jb25maWcYAyABKAsyKy5zbmFwY
+  2hhdC5yZXNlYXJjaC5nYm1sLkxvY2FsUmVzb3VyY2VDb25maWdCGuI/FxIVbG9jYWxJbmZlcmVuY2VyQ29uZmlnSABSFWxvY2FsS
+  W5mZXJlbmNlckNvbmZpZ0ITChFpbmZlcmVuY2VyX2NvbmZpZyKXCAoUU2hhcmVkUmVzb3VyY2VDb25maWcSfgoPcmVzb3VyY2Vfb
+  GFiZWxzGAEgAygLMkAuc25hcGNoYXQucmVzZWFyY2guZ2JtbC5TaGFyZWRSZXNvdXJjZUNvbmZpZy5SZXNvdXJjZUxhYmVsc0Vud
+  HJ5QhPiPxASDnJlc291cmNlTGFiZWxzUg5yZXNvdXJjZUxhYmVscxKOAQoVY29tbW9uX2NvbXB1dGVfY29uZmlnGAIgASgLMkAuc
+  25hcGNoYXQucmVzZWFyY2guZ2JtbC5TaGFyZWRSZXNvdXJjZUNvbmZpZy5Db21tb25Db21wdXRlQ29uZmlnQhjiPxUSE2NvbW1vb
+  kNvbXB1dGVDb25maWdSE2NvbW1vbkNvbXB1dGVDb25maWcalAUKE0NvbW1vbkNvbXB1dGVDb25maWcSJgoHcHJvamVjdBgBIAEoC
+  UIM4j8JEgdwcm9qZWN0Ugdwcm9qZWN0EiMKBnJlZ2lvbhgCIAEoCUIL4j8IEgZyZWdpb25SBnJlZ2lvbhJDChJ0ZW1wX2Fzc2V0c
+  19idWNrZXQYAyABKAlCFeI/EhIQdGVtcEFzc2V0c0J1Y2tldFIQdGVtcEFzc2V0c0J1Y2tldBJcCht0ZW1wX3JlZ2lvbmFsX2Fzc
+  2V0c19idWNrZXQYBCABKAlCHeI/GhIYdGVtcFJlZ2lvbmFsQXNzZXRzQnVja2V0Uhh0ZW1wUmVnaW9uYWxBc3NldHNCdWNrZXQSQ
+  woScGVybV9hc3NldHNfYnVja2V0GAUgASgJQhXiPxISEHBlcm1Bc3NldHNCdWNrZXRSEHBlcm1Bc3NldHNCdWNrZXQSWgobdGVtc
+  F9hc3NldHNfYnFfZGF0YXNldF9uYW1lGAYgASgJQhziPxkSF3RlbXBBc3NldHNCcURhdGFzZXROYW1lUhd0ZW1wQXNzZXRzQnFEY
+  XRhc2V0TmFtZRJWChllbWJlZGRpbmdfYnFfZGF0YXNldF9uYW1lGAcgASgJQhviPxgSFmVtYmVkZGluZ0JxRGF0YXNldE5hbWVSF
+  mVtYmVkZGluZ0JxRGF0YXNldE5hbWUSVgoZZ2NwX3NlcnZpY2VfYWNjb3VudF9lbWFpbBgIIAEoCUIb4j8YEhZnY3BTZXJ2aWNlQ
+  WNjb3VudEVtYWlsUhZnY3BTZXJ2aWNlQWNjb3VudEVtYWlsEjwKD2RhdGFmbG93X3J1bm5lchgLIAEoCUIT4j8QEg5kYXRhZmxvd
+  1J1bm5lclIOZGF0YWZsb3dSdW5uZXIaVwoTUmVzb3VyY2VMYWJlbHNFbnRyeRIaCgNrZXkYASABKAlCCOI/BRIDa2V5UgNrZXkSI
+  AoFdmFsdWUYAiABKAlCCuI/BxIFdmFsdWVSBXZhbHVlOgI4ASL3CAoSR2lnbFJlc291cmNlQ29uZmlnElsKGnNoYXJlZF9yZXNvd
+  XJjZV9jb25maWdfdXJpGAEgASgJQhziPxkSF3NoYXJlZFJlc291cmNlQ29uZmlnVXJpSABSF3NoYXJlZFJlc291cmNlQ29uZmlnV
+  XJpEn8KFnNoYXJlZF9yZXNvdXJjZV9jb25maWcYAiABKAsyLC5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLlNoYXJlZFJlc291cmNlQ
+  29uZmlnQhniPxYSFHNoYXJlZFJlc291cmNlQ29uZmlnSABSFHNoYXJlZFJlc291cmNlQ29uZmlnEngKE3ByZXByb2Nlc3Nvcl9jb
+  25maWcYDCABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRhdGFQcmVwcm9jZXNzb3JDb25maWdCF+I/FBIScHJlcHJvY2Vzc
+  29yQ29uZmlnUhJwcmVwcm9jZXNzb3JDb25maWcSfwoXc3ViZ3JhcGhfc2FtcGxlcl9jb25maWcYDSABKAsyKy5zbmFwY2hhdC5yZ
+  XNlYXJjaC5nYm1sLlNwYXJrUmVzb3VyY2VDb25maWdCGuI/FxIVc3ViZ3JhcGhTYW1wbGVyQ29uZmlnUhVzdWJncmFwaFNhbXBsZ
+  XJDb25maWcSfAoWc3BsaXRfZ2VuZXJhdG9yX2NvbmZpZxgOIAEoCzIrLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuU3BhcmtSZXNvd
+  XJjZUNvbmZpZ0IZ4j8WEhRzcGxpdEdlbmVyYXRvckNvbmZpZ1IUc3BsaXRHZW5lcmF0b3JDb25maWcSbQoOdHJhaW5lcl9jb25ma
+  WcYDyABKAsyMC5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRpc3RyaWJ1dGVkVHJhaW5lckNvbmZpZ0IUGAHiPw8SDXRyYWluZXJDb
+  25maWdSDXRyYWluZXJDb25maWcSdAoRaW5mZXJlbmNlcl9jb25maWcYECABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRhd
+  GFmbG93UmVzb3VyY2VDb25maWdCFxgB4j8SEhBpbmZlcmVuY2VyQ29uZmlnUhBpbmZlcmVuY2VyQ29uZmlnEoEBChd0cmFpbmVyX
+  3Jlc291cmNlX2NvbmZpZxgRIAEoCzItLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuVHJhaW5lclJlc291cmNlQ29uZmlnQhriPxcSF
+  XRyYWluZXJSZXNvdXJjZUNvbmZpZ1IVdHJhaW5lclJlc291cmNlQ29uZmlnEo0BChppbmZlcmVuY2VyX3Jlc291cmNlX2NvbmZpZ
+  xgSIAEoCzIwLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuSW5mZXJlbmNlclJlc291cmNlQ29uZmlnQh3iPxoSGGluZmVyZW5jZXJSZ
+  XNvdXJjZUNvbmZpZ1IYaW5mZXJlbmNlclJlc291cmNlQ29uZmlnQhEKD3NoYXJlZF9yZXNvdXJjZSrjAwoJQ29tcG9uZW50Ei0KE
+  UNvbXBvbmVudF9Vbmtub3duEAAaFuI/ExIRQ29tcG9uZW50X1Vua25vd24SPwoaQ29tcG9uZW50X0NvbmZpZ19WYWxpZGF0b3IQA
+  Rof4j8cEhpDb21wb25lbnRfQ29uZmlnX1ZhbGlkYXRvchI/ChpDb21wb25lbnRfQ29uZmlnX1BvcHVsYXRvchACGh/iPxwSGkNvb
+  XBvbmVudF9Db25maWdfUG9wdWxhdG9yEkEKG0NvbXBvbmVudF9EYXRhX1ByZXByb2Nlc3NvchADGiDiPx0SG0NvbXBvbmVudF9EY
+  XRhX1ByZXByb2Nlc3NvchI/ChpDb21wb25lbnRfU3ViZ3JhcGhfU2FtcGxlchAEGh/iPxwSGkNvbXBvbmVudF9TdWJncmFwaF9TY
+  W1wbGVyEj0KGUNvbXBvbmVudF9TcGxpdF9HZW5lcmF0b3IQBRoe4j8bEhlDb21wb25lbnRfU3BsaXRfR2VuZXJhdG9yEi0KEUNvb
+  XBvbmVudF9UcmFpbmVyEAYaFuI/ExIRQ29tcG9uZW50X1RyYWluZXISMwoUQ29tcG9uZW50X0luZmVyZW5jZXIQBxoZ4j8WEhRDb
+  21wb25lbnRfSW5mZXJlbmNlcmIGcHJvdG8z"""
       ).mkString)
   lazy val scalaDescriptor: _root_.scalapb.descriptors.FileDescriptor = {
     val scalaProto = com.google.protobuf.descriptor.FileDescriptorProto.parseFrom(ProtoBytes)
@@ -118,4 +119,4 @@ object GiglResourceConfigProto extends _root_.scalapb.GeneratedFileObject {
   }
   @deprecated("Use javaDescriptor instead. In a future version this will refer to scalaDescriptor.", "ScalaPB 0.5.47")
   def descriptor: com.google.protobuf.Descriptors.FileDescriptor = javaDescriptor
-}
\ No newline at end of file
+}
diff --git a/scala/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/VertexAiResourceConfig.scala b/scala/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/VertexAiResourceConfig.scala
index fbbdb297d..152f96abb 100644
--- a/scala/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/VertexAiResourceConfig.scala
+++ b/scala/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/VertexAiResourceConfig.scala
@@ -16,8 +16,13 @@ package snapchat.research.gbml.gigl_resource_config
   * @param numReplicas
   *   Num workers for job
   * @param timeout
-  *   Timeout in seconds for the job. If unset or zero, will use the default @ google.cloud.aiplatform.CustomJob, which is 7 days: 
+  *   Timeout in seconds for the job. If unset or zero, will use the default @ google.cloud.aiplatform.CustomJob, which is 7 days:
   *   https://github.com/googleapis/python-aiplatform/blob/58fbabdeeefd1ccf1a9d0c22eeb5606aeb9c2266/google/cloud/aiplatform/jobs.py#L2252-L2253
+  * @param gcpRegionOverride
+  *   Region override.
+  *   If provided, then the Vertex AI Job will be launched in the provided region.
+  *   ex: "us-west1"
+  *   NOTE: If set, then there may be data egress costs from CommonComputeConfig.region -> gcp_region_override
   */
 @SerialVersionUID(0L)
 final case class VertexAiResourceConfig(
@@ -26,47 +31,55 @@ final case class VertexAiResourceConfig(
     gpuLimit: _root_.scala.Int = 0,
     numReplicas: _root_.scala.Int = 0,
     timeout: _root_.scala.Int = 0,
+    gcpRegionOverride: _root_.scala.Predef.String = "",
     unknownFields: _root_.scalapb.UnknownFieldSet = _root_.scalapb.UnknownFieldSet.empty
     ) extends scalapb.GeneratedMessage with scalapb.lenses.Updatable[VertexAiResourceConfig] {
     @transient
     private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
     private[this] def __computeSerializedSize(): _root_.scala.Int = {
       var __size = 0
-      
+
       {
         val __value = machineType
         if (!__value.isEmpty) {
           __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
         }
       };
-      
+
       {
         val __value = gpuType
         if (!__value.isEmpty) {
           __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(2, __value)
         }
       };
-      
+
       {
         val __value = gpuLimit
         if (__value != 0) {
           __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(3, __value)
         }
       };
-      
+
       {
         val __value = numReplicas
         if (__value != 0) {
           __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(4, __value)
         }
       };
-      
+
       {
         val __value = timeout
         if (__value != 0) {
           __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(5, __value)
         }
       };
+
+      {
+        val __value = gcpRegionOverride
+        if (!__value.isEmpty) {
+          __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(6, __value)
+        }
+      };
       __size += unknownFields.serializedSize
       __size
     }
@@ -77,7 +90,7 @@ final case class VertexAiResourceConfig(
         __serializedSizeMemoized = __size
       }
       __size - 1
-      
+
     }
     def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
       {
@@ -110,6 +123,12 @@ final case class VertexAiResourceConfig(
           _output__.writeUInt32(5, __v)
         }
       };
+      {
+        val __v = gcpRegionOverride
+        if (!__v.isEmpty) {
+          _output__.writeString(6, __v)
+        }
+      };
       unknownFields.writeTo(_output__)
     }
     def withMachineType(__v: _root_.scala.Predef.String): VertexAiResourceConfig = copy(machineType = __v)
@@ -117,6 +136,7 @@ final case class VertexAiResourceConfig(
     def withGpuLimit(__v: _root_.scala.Int): VertexAiResourceConfig = copy(gpuLimit = __v)
     def withNumReplicas(__v: _root_.scala.Int): VertexAiResourceConfig = copy(numReplicas = __v)
     def withTimeout(__v: _root_.scala.Int): VertexAiResourceConfig = copy(timeout = __v)
+    def withGcpRegionOverride(__v: _root_.scala.Predef.String): VertexAiResourceConfig = copy(gcpRegionOverride = __v)
     def withUnknownFields(__v: _root_.scalapb.UnknownFieldSet) = copy(unknownFields = __v)
     def discardUnknownFields = copy(unknownFields = _root_.scalapb.UnknownFieldSet.empty)
     def getFieldByNumber(__fieldNumber: _root_.scala.Int): _root_.scala.Any = {
@@ -141,6 +161,10 @@ final case class VertexAiResourceConfig(
           val __t = timeout
           if (__t != 0) __t else null
         }
+        case 6 => {
+          val __t = gcpRegionOverride
+          if (__t != "") __t else null
+        }
       }
     }
     def getField(__field: _root_.scalapb.descriptors.FieldDescriptor): _root_.scalapb.descriptors.PValue = {
@@ -151,6 +175,7 @@ final case class VertexAiResourceConfig(
         case 3 => _root_.scalapb.descriptors.PInt(gpuLimit)
         case 4 => _root_.scalapb.descriptors.PInt(numReplicas)
         case 5 => _root_.scalapb.descriptors.PInt(timeout)
+        case 6 => _root_.scalapb.descriptors.PString(gcpRegionOverride)
       }
     }
     def toProtoString: _root_.scala.Predef.String = _root_.scalapb.TextFormat.printToUnicodeString(this)
@@ -166,6 +191,7 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
     var __gpuLimit: _root_.scala.Int = 0
     var __numReplicas: _root_.scala.Int = 0
     var __timeout: _root_.scala.Int = 0
+    var __gcpRegionOverride: _root_.scala.Predef.String = ""
     var `_unknownFields__`: _root_.scalapb.UnknownFieldSet.Builder = null
     var _done__ = false
     while (!_done__) {
@@ -182,6 +208,8 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
           __numReplicas = _input__.readUInt32()
         case 40 =>
           __timeout = _input__.readUInt32()
+        case 50 =>
+          __gcpRegionOverride = _input__.readStringRequireUtf8()
         case tag =>
           if (_unknownFields__ == null) {
             _unknownFields__ = new _root_.scalapb.UnknownFieldSet.Builder()
@@ -195,6 +223,7 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
         gpuLimit = __gpuLimit,
         numReplicas = __numReplicas,
         timeout = __timeout,
+        gcpRegionOverride = __gcpRegionOverride,
         unknownFields = if (_unknownFields__ == null) _root_.scalapb.UnknownFieldSet.empty else _unknownFields__.result()
     )
   }
@@ -206,7 +235,8 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
         gpuType = __fieldsMap.get(scalaDescriptor.findFieldByNumber(2).get).map(_.as[_root_.scala.Predef.String]).getOrElse(""),
         gpuLimit = __fieldsMap.get(scalaDescriptor.findFieldByNumber(3).get).map(_.as[_root_.scala.Int]).getOrElse(0),
         numReplicas = __fieldsMap.get(scalaDescriptor.findFieldByNumber(4).get).map(_.as[_root_.scala.Int]).getOrElse(0),
-        timeout = __fieldsMap.get(scalaDescriptor.findFieldByNumber(5).get).map(_.as[_root_.scala.Int]).getOrElse(0)
+        timeout = __fieldsMap.get(scalaDescriptor.findFieldByNumber(5).get).map(_.as[_root_.scala.Int]).getOrElse(0),
+        gcpRegionOverride = __fieldsMap.get(scalaDescriptor.findFieldByNumber(6).get).map(_.as[_root_.scala.Predef.String]).getOrElse("")
       )
     case _ => throw new RuntimeException("Expected PMessage")
   }
@@ -220,7 +250,8 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
     gpuType = "",
     gpuLimit = 0,
     numReplicas = 0,
-    timeout = 0
+    timeout = 0,
+    gcpRegionOverride = ""
   )
   implicit class VertexAiResourceConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gigl_resource_config.VertexAiResourceConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gigl_resource_config.VertexAiResourceConfig](_l) {
     def machineType: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.machineType)((c_, f_) => c_.copy(machineType = f_))
@@ -228,24 +259,28 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
     def gpuLimit: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Int] = field(_.gpuLimit)((c_, f_) => c_.copy(gpuLimit = f_))
     def numReplicas: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Int] = field(_.numReplicas)((c_, f_) => c_.copy(numReplicas = f_))
     def timeout: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Int] = field(_.timeout)((c_, f_) => c_.copy(timeout = f_))
+    def gcpRegionOverride: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.gcpRegionOverride)((c_, f_) => c_.copy(gcpRegionOverride = f_))
   }
   final val MACHINE_TYPE_FIELD_NUMBER = 1
   final val GPU_TYPE_FIELD_NUMBER = 2
   final val GPU_LIMIT_FIELD_NUMBER = 3
   final val NUM_REPLICAS_FIELD_NUMBER = 4
   final val TIMEOUT_FIELD_NUMBER = 5
+  final val GCP_REGION_OVERRIDE_FIELD_NUMBER = 6
   def of(
     machineType: _root_.scala.Predef.String,
     gpuType: _root_.scala.Predef.String,
     gpuLimit: _root_.scala.Int,
     numReplicas: _root_.scala.Int,
-    timeout: _root_.scala.Int
+    timeout: _root_.scala.Int,
+    gcpRegionOverride: _root_.scala.Predef.String
   ): _root_.snapchat.research.gbml.gigl_resource_config.VertexAiResourceConfig = _root_.snapchat.research.gbml.gigl_resource_config.VertexAiResourceConfig(
     machineType,
     gpuType,
     gpuLimit,
     numReplicas,
-    timeout
+    timeout,
+    gcpRegionOverride
   )
   // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.VertexAiResourceConfig])
 }
diff --git a/scala_spark35/common/src/main/scala/snapchat/research/gbml/gbml_config/GbmlConfig.scala b/scala_spark35/common/src/main/scala/snapchat/research/gbml/gbml_config/GbmlConfig.scala
index 12adb5a17..b3f110643 100644
--- a/scala_spark35/common/src/main/scala/snapchat/research/gbml/gbml_config/GbmlConfig.scala
+++ b/scala_spark35/common/src/main/scala/snapchat/research/gbml/gbml_config/GbmlConfig.scala
@@ -73,7 +73,7 @@ final case class GbmlConfig(
         __serializedSizeMemoized = __size
       }
       __size - 1
-      
+
     }
     def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
       taskMetadata.foreach { __v =>
@@ -359,7 +359,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         taskMetadata.nodeBasedTaskMetadata.foreach { __v =>
@@ -411,7 +411,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.type = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TaskMetadata])
   }
-  
+
   object TaskMetadata extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata = {
@@ -491,7 +491,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         override def number: _root_.scala.Int = 0
         override def value: _root_.scala.Nothing = throw new java.util.NoSuchElementException("Empty.value")
       }
-    
+
       @SerialVersionUID(0L)
       final case class NodeBasedTaskMetadata(value: snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata) extends snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.TaskMetadata {
         type ValueType = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata
@@ -537,7 +537,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           supervisionNodeTypes.foreach { __v =>
@@ -567,7 +567,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata.type = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata])
     }
-    
+
     object NodeBasedTaskMetadata extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata = {
@@ -619,7 +619,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata])
     }
-    
+
     @SerialVersionUID(0L)
     final case class NodeAnchorBasedLinkPredictionTaskMetadata(
         supervisionEdgeTypes: _root_.scala.Seq[snapchat.research.gbml.graph_schema.EdgeType] = _root_.scala.Seq.empty,
@@ -643,7 +643,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           supervisionEdgeTypes.foreach { __v =>
@@ -675,7 +675,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata.type = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata])
     }
-    
+
     object NodeAnchorBasedLinkPredictionTaskMetadata extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata = {
@@ -733,7 +733,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata])
     }
-    
+
     @SerialVersionUID(0L)
     final case class LinkBasedTaskMetadata(
         supervisionEdgeTypes: _root_.scala.Seq[snapchat.research.gbml.graph_schema.EdgeType] = _root_.scala.Seq.empty,
@@ -757,7 +757,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           supervisionEdgeTypes.foreach { __v =>
@@ -789,7 +789,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata.type = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata])
     }
-    
+
     object LinkBasedTaskMetadata extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata = {
@@ -847,7 +847,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata])
     }
-    
+
     implicit class TaskMetadataLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata](_l) {
       def nodeBasedTaskMetadata: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata] = field(_.getNodeBasedTaskMetadata)((c_, f_) => c_.copy(taskMetadata = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.TaskMetadata.NodeBasedTaskMetadata(f_)))
       def nodeAnchorBasedLinkPredictionTaskMetadata: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata] = field(_.getNodeAnchorBasedLinkPredictionTaskMetadata)((c_, f_) => c_.copy(taskMetadata = snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata(f_)))
@@ -864,7 +864,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TaskMetadata])
   }
-  
+
   /** @param preprocessedMetadataUri
     *   Uri where DataPreprocessor generates the PreprocessedMetadata proto.
     * @param flattenedGraphMetadata
@@ -917,7 +917,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = preprocessedMetadataUri
           if (!__value.isEmpty) {
@@ -948,42 +948,42 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           val __value = snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig._typemapper_sharedArgs.toBase(__item)
           __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
         }
-        
+
         {
           val __value = isGraphDirected
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(7, __value)
           }
         };
-        
+
         {
           val __value = shouldSkipTraining
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(8, __value)
           }
         };
-        
+
         {
           val __value = shouldSkipAutomaticTempAssetCleanup
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(9, __value)
           }
         };
-        
+
         {
           val __value = shouldSkipInference
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(10, __value)
           }
         };
-        
+
         {
           val __value = shouldSkipModelEvaluation
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(11, __value)
           }
         };
-        
+
         {
           val __value = shouldIncludeIsolatedNodesInTraining
           if (__value != false) {
@@ -1000,7 +1000,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -1171,7 +1171,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.SharedConfig])
   }
-  
+
   object SharedConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig = {
@@ -1308,14 +1308,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -1332,7 +1332,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -1376,7 +1376,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.SharedArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.SharedArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.SharedConfig.SharedArgsEntry])
     }
-    
+
     object SharedArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.SharedArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.SharedArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig.SharedArgsEntry = {
@@ -1441,7 +1441,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.SharedConfig.SharedArgsEntry])
     }
-    
+
     implicit class SharedConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.SharedConfig](_l) {
       def preprocessedMetadataUri: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.preprocessedMetadataUri)((c_, f_) => c_.copy(preprocessedMetadataUri = f_))
       def flattenedGraphMetadata: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.flattened_graph_metadata.FlattenedGraphMetadata] = field(_.getFlattenedGraphMetadata)((c_, f_) => c_.copy(flattenedGraphMetadata = Option(f_)))
@@ -1508,7 +1508,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.SharedConfig])
   }
-  
+
   /** Contains config related to generating training data for a GML task.
     */
   @SerialVersionUID(0L)
@@ -1544,7 +1544,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         dataPreprocessorConfig.foreach { __v =>
@@ -1597,7 +1597,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig])
   }
-  
+
   object DatasetConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig = {
@@ -1678,7 +1678,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = dataPreprocessorConfigClsPath
             if (!__value.isEmpty) {
@@ -1699,7 +1699,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -1743,7 +1743,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.DataPreprocessorConfig])
     }
-    
+
     object DataPreprocessorConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig = {
@@ -1809,14 +1809,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
           private[this] def __computeSerializedSize(): _root_.scala.Int = {
             var __size = 0
-            
+
             {
               val __value = key
               if (!__value.isEmpty) {
                 __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
               }
             };
-            
+
             {
               val __value = value
               if (!__value.isEmpty) {
@@ -1833,7 +1833,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
               __serializedSizeMemoized = __size
             }
             __size - 1
-            
+
           }
           def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
             {
@@ -1877,7 +1877,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry
           // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry])
       }
-      
+
       object DataPreprocessorArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry] {
         implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry] = this
         def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry = {
@@ -1942,7 +1942,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         )
         // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry])
       }
-      
+
       implicit class DataPreprocessorConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig](_l) {
         def dataPreprocessorConfigClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.dataPreprocessorConfigClsPath)((c_, f_) => c_.copy(dataPreprocessorConfigClsPath = f_))
         def dataPreprocessorArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.dataPreprocessorArgs)((c_, f_) => c_.copy(dataPreprocessorArgs = f_))
@@ -1960,7 +1960,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.DataPreprocessorConfig])
     }
-    
+
     /** @param numHops
       *   number of hops for subgraph sampler to include
       * @param numNeighborsToSample
@@ -1990,7 +1990,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       *   Also used in loss computation. Random negatives will always be used even when there
       *   are no user defined hard negatives
       * @param graphDbConfig
-      *   If specified, intention is to run ingestion into graphDB for subgraph sampler 
+      *   If specified, intention is to run ingestion into graphDB for subgraph sampler
       */
     @SerialVersionUID(0L)
     final case class SubgraphSamplerConfig(
@@ -2009,14 +2009,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = numHops
             if (__value != 0) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(1, __value)
             }
           };
-          
+
           {
             val __value = numNeighborsToSample
             if (__value != 0) {
@@ -2027,7 +2027,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             val __value = subgraphSamplingStrategy.get
             __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
           };
-          
+
           {
             val __value = numPositiveSamples
             if (__value != 0) {
@@ -2038,21 +2038,21 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             val __value = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig._typemapper_experimentalFlags.toBase(__item)
             __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
           }
-          
+
           {
             val __value = numMaxTrainingSamplesToOutput
             if (__value != 0) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(6, __value)
             }
           };
-          
+
           {
             val __value = numUserDefinedPositiveSamples
             if (__value != 0) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(7, __value)
             }
           };
-          
+
           {
             val __value = numUserDefinedNegativeSamples
             if (__value != 0) {
@@ -2073,7 +2073,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -2199,7 +2199,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.SubgraphSamplerConfig])
     }
-    
+
     object SubgraphSamplerConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig = {
@@ -2309,14 +2309,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
           private[this] def __computeSerializedSize(): _root_.scala.Int = {
             var __size = 0
-            
+
             {
               val __value = key
               if (!__value.isEmpty) {
                 __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
               }
             };
-            
+
             {
               val __value = value
               if (!__value.isEmpty) {
@@ -2333,7 +2333,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
               __serializedSizeMemoized = __size
             }
             __size - 1
-            
+
           }
           def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
             {
@@ -2377,7 +2377,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry
           // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry])
       }
-      
+
       object ExperimentalFlagsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry] {
         implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry] = this
         def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry = {
@@ -2442,7 +2442,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         )
         // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry])
       }
-      
+
       implicit class SubgraphSamplerConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SubgraphSamplerConfig](_l) {
         def numHops: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Int] = field(_.numHops)((c_, f_) => c_.copy(numHops = f_))
         def numNeighborsToSample: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Int] = field(_.numNeighborsToSample)((c_, f_) => c_.copy(numNeighborsToSample = f_))
@@ -2490,7 +2490,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.SubgraphSamplerConfig])
     }
-    
+
     /** @param splitStrategyClsPath
       *   Module path to concrete SplitStrategy instance.
       * @param splitStrategyArgs
@@ -2512,7 +2512,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = splitStrategyClsPath
             if (!__value.isEmpty) {
@@ -2523,7 +2523,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             val __value = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig._typemapper_splitStrategyArgs.toBase(__item)
             __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
           }
-          
+
           {
             val __value = assignerClsPath
             if (!__value.isEmpty) {
@@ -2544,7 +2544,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -2612,7 +2612,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig])
     }
-    
+
     object SplitGeneratorConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig = {
@@ -2692,14 +2692,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
           private[this] def __computeSerializedSize(): _root_.scala.Int = {
             var __size = 0
-            
+
             {
               val __value = key
               if (!__value.isEmpty) {
                 __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
               }
             };
-            
+
             {
               val __value = value
               if (!__value.isEmpty) {
@@ -2716,7 +2716,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
               __serializedSizeMemoized = __size
             }
             __size - 1
-            
+
           }
           def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
             {
@@ -2760,7 +2760,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry
           // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry])
       }
-      
+
       object SplitStrategyArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry] {
         implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry] = this
         def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry = {
@@ -2825,7 +2825,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         )
         // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry])
       }
-      
+
       @SerialVersionUID(0L)
       final case class AssignerArgsEntry(
           key: _root_.scala.Predef.String = "",
@@ -2836,14 +2836,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
           private[this] def __computeSerializedSize(): _root_.scala.Int = {
             var __size = 0
-            
+
             {
               val __value = key
               if (!__value.isEmpty) {
                 __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
               }
             };
-            
+
             {
               val __value = value
               if (!__value.isEmpty) {
@@ -2860,7 +2860,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
               __serializedSizeMemoized = __size
             }
             __size - 1
-            
+
           }
           def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
             {
@@ -2904,7 +2904,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           def companion: snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry
           // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry])
       }
-      
+
       object AssignerArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry] {
         implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry] = this
         def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry = {
@@ -2969,7 +2969,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         )
         // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry])
       }
-      
+
       implicit class SplitGeneratorConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.SplitGeneratorConfig](_l) {
         def splitStrategyClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.splitStrategyClsPath)((c_, f_) => c_.copy(splitStrategyClsPath = f_))
         def splitStrategyArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.splitStrategyArgs)((c_, f_) => c_.copy(splitStrategyArgs = f_))
@@ -2997,7 +2997,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig.SplitGeneratorConfig])
     }
-    
+
     implicit class DatasetConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig](_l) {
       def dataPreprocessorConfig: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig] = field(_.getDataPreprocessorConfig)((c_, f_) => c_.copy(dataPreprocessorConfig = Option(f_)))
       def optionalDataPreprocessorConfig: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Option[snapchat.research.gbml.gbml_config.GbmlConfig.DatasetConfig.DataPreprocessorConfig]] = field(_.dataPreprocessorConfig)((c_, f_) => c_.copy(dataPreprocessorConfig = f_))
@@ -3020,11 +3020,11 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.DatasetConfig])
   }
-  
+
   /** Generic Configuration for a GraphDB connection.
     *
     * @param graphDbIngestionClsPath
-    *   Python class path pointing to user-written 
+    *   Python class path pointing to user-written
     *   `BaseIngestion`` class definition. e.g. `my.team.graph_db.BaseInjectionImpl`.
     *   This class is currently, as an implementation detail, used for injestion only.
     *   We document this *purely* for information purposes and may change the implementation at any time.
@@ -3052,7 +3052,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = graphDbIngestionClsPath
           if (!__value.isEmpty) {
@@ -3081,7 +3081,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -3148,7 +3148,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.GraphDBConfig])
   }
-  
+
   object GraphDBConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig = {
@@ -3230,14 +3230,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -3254,7 +3254,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -3298,7 +3298,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry])
     }
-    
+
     object GraphDbIngestionArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry = {
@@ -3363,7 +3363,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry])
     }
-    
+
     @SerialVersionUID(0L)
     final case class GraphDbArgsEntry(
         key: _root_.scala.Predef.String = "",
@@ -3374,14 +3374,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -3398,7 +3398,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -3442,7 +3442,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDbArgsEntry])
     }
-    
+
     object GraphDbArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDbArgsEntry = {
@@ -3507,7 +3507,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDbArgsEntry])
     }
-    
+
     /** Scala-specific configuration.
       *
       * @param graphDbClientClassPath
@@ -3523,7 +3523,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = graphDbClientClassPath
             if (!__value.isEmpty) {
@@ -3540,7 +3540,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -3572,7 +3572,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDBServiceConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDBServiceConfig
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDBServiceConfig])
     }
-    
+
     object GraphDBServiceConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDBServiceConfig] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDBServiceConfig] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig.GraphDBServiceConfig = {
@@ -3624,7 +3624,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.GraphDBConfig.GraphDBServiceConfig])
     }
-    
+
     implicit class GraphDBConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.GraphDBConfig](_l) {
       def graphDbIngestionClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.graphDbIngestionClsPath)((c_, f_) => c_.copy(graphDbIngestionClsPath = f_))
       def graphDbIngestionArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.graphDbIngestionArgs)((c_, f_) => c_.copy(graphDbIngestionArgs = f_))
@@ -3653,10 +3653,10 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.GraphDBConfig])
   }
-  
+
   /** @param trainerClsPath
     *   (deprecated)
-    *   Uri pointing to user-written BaseTrainer class definition. Used for the subgraph-sampling-based training process. 
+    *   Uri pointing to user-written BaseTrainer class definition. Used for the subgraph-sampling-based training process.
     * @param trainerArgs
     *   Arguments to parameterize training process with.
     * @param shouldLogToTensorboard
@@ -3674,7 +3674,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = trainerClsPath
           if (!__value.isEmpty) {
@@ -3693,7 +3693,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           val __value = executable.command.get
           __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(101, __value)
         };
-        
+
         {
           val __value = shouldLogToTensorboard
           if (__value != false) {
@@ -3710,7 +3710,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -3784,7 +3784,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TrainerConfig])
   }
-  
+
   object TrainerConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig = {
@@ -3873,7 +3873,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         override def number: _root_.scala.Int = 0
         override def value: _root_.scala.Nothing = throw new java.util.NoSuchElementException("Empty.value")
       }
-    
+
       @SerialVersionUID(0L)
       final case class ClsPath(value: _root_.scala.Predef.String) extends snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.Executable {
         type ValueType = _root_.scala.Predef.String
@@ -3899,14 +3899,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -3923,7 +3923,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -3967,7 +3967,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.TrainerArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.TrainerArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.TrainerConfig.TrainerArgsEntry])
     }
-    
+
     object TrainerArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.TrainerArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.TrainerArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig.TrainerArgsEntry = {
@@ -4032,7 +4032,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TrainerConfig.TrainerArgsEntry])
     }
-    
+
     implicit class TrainerConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TrainerConfig](_l) {
       def trainerClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.trainerClsPath)((c_, f_) => c_.copy(trainerClsPath = f_))
       def trainerArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.trainerArgs)((c_, f_) => c_.copy(trainerArgs = f_))
@@ -4061,10 +4061,10 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.TrainerConfig])
   }
-  
+
   /** @param inferencerClsPath
     *   (deprecated)
-    *   Path to modeling task spec class path to construct model for inference. Used for the subgraph-sampling-based inference process. 
+    *   Path to modeling task spec class path to construct model for inference. Used for the subgraph-sampling-based inference process.
     * @param inferenceBatchSize
     *   Optional. If set, will be used to batch inference samples to a specific size before call for inference is made
     *   Defaults to setting in python/gigl/src/inference/gnn_inferencer.py
@@ -4085,7 +4085,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           val __value = snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig._typemapper_inferencerArgs.toBase(__item)
           __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
         }
-        
+
         {
           val __value = inferencerClsPath
           if (!__value.isEmpty) {
@@ -4100,7 +4100,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           val __value = executable.command.get
           __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(101, __value)
         };
-        
+
         {
           val __value = inferenceBatchSize
           if (__value != 0) {
@@ -4117,7 +4117,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         inferencerArgs.foreach { __v =>
@@ -4191,7 +4191,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.InferencerConfig])
   }
-  
+
   object InferencerConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig = {
@@ -4280,7 +4280,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         override def number: _root_.scala.Int = 0
         override def value: _root_.scala.Nothing = throw new java.util.NoSuchElementException("Empty.value")
       }
-    
+
       @SerialVersionUID(0L)
       final case class ClsPath(value: _root_.scala.Predef.String) extends snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.Executable {
         type ValueType = _root_.scala.Predef.String
@@ -4306,14 +4306,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -4330,7 +4330,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -4374,7 +4374,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.InferencerArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.InferencerArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.InferencerConfig.InferencerArgsEntry])
     }
-    
+
     object InferencerArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.InferencerArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.InferencerArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig.InferencerArgsEntry = {
@@ -4439,7 +4439,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.InferencerConfig.InferencerArgsEntry])
     }
-    
+
     implicit class InferencerConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.InferencerConfig](_l) {
       def inferencerArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.inferencerArgs)((c_, f_) => c_.copy(inferencerArgs = f_))
       def inferencerClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.inferencerClsPath)((c_, f_) => c_.copy(inferencerClsPath = f_))
@@ -4468,7 +4468,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.InferencerConfig])
   }
-  
+
   @SerialVersionUID(0L)
   final case class PostProcessorConfig(
       postProcessorArgs: _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String] = _root_.scala.collection.immutable.Map.empty,
@@ -4483,7 +4483,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           val __value = snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig._typemapper_postProcessorArgs.toBase(__item)
           __size += 1 + _root_.com.google.protobuf.CodedOutputStream.computeUInt32SizeNoTag(__value.serializedSize) + __value.serializedSize
         }
-        
+
         {
           val __value = postProcessorClsPath
           if (!__value.isEmpty) {
@@ -4500,7 +4500,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         postProcessorArgs.foreach { __v =>
@@ -4544,7 +4544,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.PostProcessorConfig])
   }
-  
+
   object PostProcessorConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig = {
@@ -4610,14 +4610,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -4634,7 +4634,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -4678,7 +4678,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry])
     }
-    
+
     object PostProcessorArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry = {
@@ -4743,7 +4743,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry])
     }
-    
+
     implicit class PostProcessorConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.PostProcessorConfig](_l) {
       def postProcessorArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.postProcessorArgs)((c_, f_) => c_.copy(postProcessorArgs = f_))
       def postProcessorClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.postProcessorClsPath)((c_, f_) => c_.copy(postProcessorClsPath = f_))
@@ -4761,7 +4761,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.PostProcessorConfig])
   }
-  
+
   @SerialVersionUID(0L)
   final case class MetricsConfig(
       metricsClsPath: _root_.scala.Predef.String = "",
@@ -4772,7 +4772,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = metricsClsPath
           if (!__value.isEmpty) {
@@ -4793,7 +4793,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -4837,7 +4837,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.MetricsConfig])
   }
-  
+
   object MetricsConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig = {
@@ -4903,14 +4903,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -4927,7 +4927,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -4971,7 +4971,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.MetricsArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.MetricsArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.MetricsConfig.MetricsArgsEntry])
     }
-    
+
     object MetricsArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.MetricsArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.MetricsArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig.MetricsArgsEntry = {
@@ -5036,7 +5036,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.MetricsConfig.MetricsArgsEntry])
     }
-    
+
     implicit class MetricsConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.MetricsConfig](_l) {
       def metricsClsPath: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.metricsClsPath)((c_, f_) => c_.copy(metricsClsPath = f_))
       def metricsArgs: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.collection.immutable.Map[_root_.scala.Predef.String, _root_.scala.Predef.String]] = field(_.metricsArgs)((c_, f_) => c_.copy(metricsArgs = f_))
@@ -5054,7 +5054,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.MetricsConfig])
   }
-  
+
   @SerialVersionUID(0L)
   final case class ProfilerConfig(
       shouldEnableProfiler: _root_.scala.Boolean = false,
@@ -5066,14 +5066,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = shouldEnableProfiler
           if (__value != false) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeBoolSize(1, __value)
           }
         };
-        
+
         {
           val __value = profilerLogDir
           if (!__value.isEmpty) {
@@ -5094,7 +5094,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -5150,7 +5150,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.type = snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.ProfilerConfig])
   }
-  
+
   object ProfilerConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig = {
@@ -5222,14 +5222,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
         private[this] def __computeSerializedSize(): _root_.scala.Int = {
           var __size = 0
-          
+
           {
             val __value = key
             if (!__value.isEmpty) {
               __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
             }
           };
-          
+
           {
             val __value = value
             if (!__value.isEmpty) {
@@ -5246,7 +5246,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
             __serializedSizeMemoized = __size
           }
           __size - 1
-          
+
         }
         def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
           {
@@ -5290,7 +5290,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
         def companion: snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.ProfilerArgsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.ProfilerArgsEntry
         // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.ProfilerConfig.ProfilerArgsEntry])
     }
-    
+
     object ProfilerArgsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.ProfilerArgsEntry] {
       implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.ProfilerArgsEntry] = this
       def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig.ProfilerArgsEntry = {
@@ -5355,7 +5355,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       )
       // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.ProfilerConfig.ProfilerArgsEntry])
     }
-    
+
     implicit class ProfilerConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.ProfilerConfig](_l) {
       def shouldEnableProfiler: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Boolean] = field(_.shouldEnableProfiler)((c_, f_) => c_.copy(shouldEnableProfiler = f_))
       def profilerLogDir: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.profilerLogDir)((c_, f_) => c_.copy(profilerLogDir = f_))
@@ -5377,7 +5377,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.ProfilerConfig])
   }
-  
+
   @SerialVersionUID(0L)
   final case class FeatureFlagsEntry(
       key: _root_.scala.Predef.String = "",
@@ -5388,14 +5388,14 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
       private[this] def __computeSerializedSize(): _root_.scala.Int = {
         var __size = 0
-        
+
         {
           val __value = key
           if (!__value.isEmpty) {
             __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
           }
         };
-        
+
         {
           val __value = value
           if (!__value.isEmpty) {
@@ -5412,7 +5412,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
           __serializedSizeMemoized = __size
         }
         __size - 1
-        
+
       }
       def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
         {
@@ -5456,7 +5456,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
       def companion: snapchat.research.gbml.gbml_config.GbmlConfig.FeatureFlagsEntry.type = snapchat.research.gbml.gbml_config.GbmlConfig.FeatureFlagsEntry
       // @@protoc_insertion_point(GeneratedMessage[snapchat.research.gbml.GbmlConfig.FeatureFlagsEntry])
   }
-  
+
   object FeatureFlagsEntry extends scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.FeatureFlagsEntry] {
     implicit def messageCompanion: scalapb.GeneratedMessageCompanion[snapchat.research.gbml.gbml_config.GbmlConfig.FeatureFlagsEntry] = this
     def parseFrom(`_input__`: _root_.com.google.protobuf.CodedInputStream): snapchat.research.gbml.gbml_config.GbmlConfig.FeatureFlagsEntry = {
@@ -5521,7 +5521,7 @@ object GbmlConfig extends scalapb.GeneratedMessageCompanion[snapchat.research.gb
     )
     // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.GbmlConfig.FeatureFlagsEntry])
   }
-  
+
   implicit class GbmlConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig](_l) {
     def taskMetadata: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata] = field(_.getTaskMetadata)((c_, f_) => c_.copy(taskMetadata = Option(f_)))
     def optionalTaskMetadata: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Option[snapchat.research.gbml.gbml_config.GbmlConfig.TaskMetadata]] = field(_.taskMetadata)((c_, f_) => c_.copy(taskMetadata = f_))
diff --git a/scala_spark35/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/GiglResourceConfigProto.scala b/scala_spark35/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/GiglResourceConfigProto.scala
index c03fc503b..55b0dfcf7 100644
--- a/scala_spark35/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/GiglResourceConfigProto.scala
+++ b/scala_spark35/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/GiglResourceConfigProto.scala
@@ -44,68 +44,69 @@ object GiglResourceConfigProto extends _root_.scalapb.GeneratedFileObject {
   W1vcnlSZXF1ZXN0Ug1tZW1vcnlSZXF1ZXN0EicKCGdwdV90eXBlGAMgASgJQgziPwkSB2dwdVR5cGVSB2dwdVR5cGUSKgoJZ3B1X
   2xpbWl0GAQgASgNQg3iPwoSCGdwdUxpbWl0UghncHVMaW1pdBIzCgxudW1fcmVwbGljYXMYBSABKA1CEOI/DRILbnVtUmVwbGljY
   XNSC251bVJlcGxpY2FzIkYKEkxvY2FsVHJhaW5lckNvbmZpZxIwCgtudW1fd29ya2VycxgBIAEoDUIP4j8MEgpudW1Xb3JrZXJzU
-  gpudW1Xb3JrZXJzIv8BChZWZXJ0ZXhBaVJlc291cmNlQ29uZmlnEjMKDG1hY2hpbmVfdHlwZRgBIAEoCUIQ4j8NEgttYWNoaW5lV
+  gpudW1Xb3JrZXJzIscCChZWZXJ0ZXhBaVJlc291cmNlQ29uZmlnEjMKDG1hY2hpbmVfdHlwZRgBIAEoCUIQ4j8NEgttYWNoaW5lV
   HlwZVILbWFjaGluZVR5cGUSJwoIZ3B1X3R5cGUYAiABKAlCDOI/CRIHZ3B1VHlwZVIHZ3B1VHlwZRIqCglncHVfbGltaXQYAyABK
   A1CDeI/ChIIZ3B1TGltaXRSCGdwdUxpbWl0EjMKDG51bV9yZXBsaWNhcxgEIAEoDUIQ4j8NEgtudW1SZXBsaWNhc1ILbnVtUmVwb
-  GljYXMSJgoHdGltZW91dBgFIAEoDUIM4j8JEgd0aW1lb3V0Ugd0aW1lb3V0IooCChFLRlBSZXNvdXJjZUNvbmZpZxIwCgtjcHVfc
-  mVxdWVzdBgBIAEoCUIP4j8MEgpjcHVSZXF1ZXN0UgpjcHVSZXF1ZXN0EjkKDm1lbW9yeV9yZXF1ZXN0GAIgASgJQhLiPw8SDW1lb
-  W9yeVJlcXVlc3RSDW1lbW9yeVJlcXVlc3QSJwoIZ3B1X3R5cGUYAyABKAlCDOI/CRIHZ3B1VHlwZVIHZ3B1VHlwZRIqCglncHVfb
-  GltaXQYBCABKA1CDeI/ChIIZ3B1TGltaXRSCGdwdUxpbWl0EjMKDG51bV9yZXBsaWNhcxgFIAEoDUIQ4j8NEgtudW1SZXBsaWNhc
-  1ILbnVtUmVwbGljYXMiRwoTTG9jYWxSZXNvdXJjZUNvbmZpZxIwCgtudW1fd29ya2VycxgBIAEoDUIP4j8MEgpudW1Xb3JrZXJzU
-  gpudW1Xb3JrZXJzIp0DChhEaXN0cmlidXRlZFRyYWluZXJDb25maWcShAEKGHZlcnRleF9haV90cmFpbmVyX2NvbmZpZxgBIAEoC
-  zItLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuVmVydGV4QWlUcmFpbmVyQ29uZmlnQhriPxcSFXZlcnRleEFpVHJhaW5lckNvbmZpZ
-  0gAUhV2ZXJ0ZXhBaVRyYWluZXJDb25maWcSbwoSa2ZwX3RyYWluZXJfY29uZmlnGAIgASgLMiguc25hcGNoYXQucmVzZWFyY2guZ
-  2JtbC5LRlBUcmFpbmVyQ29uZmlnQhXiPxISEGtmcFRyYWluZXJDb25maWdIAFIQa2ZwVHJhaW5lckNvbmZpZxJ3ChRsb2NhbF90c
-  mFpbmVyX2NvbmZpZxgDIAEoCzIqLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuTG9jYWxUcmFpbmVyQ29uZmlnQhfiPxQSEmxvY2FsV
-  HJhaW5lckNvbmZpZ0gAUhJsb2NhbFRyYWluZXJDb25maWdCEAoOdHJhaW5lcl9jb25maWcinQMKFVRyYWluZXJSZXNvdXJjZUNvb
-  mZpZxKFAQoYdmVydGV4X2FpX3RyYWluZXJfY29uZmlnGAEgASgLMi4uc25hcGNoYXQucmVzZWFyY2guZ2JtbC5WZXJ0ZXhBaVJlc
-  291cmNlQ29uZmlnQhriPxcSFXZlcnRleEFpVHJhaW5lckNvbmZpZ0gAUhV2ZXJ0ZXhBaVRyYWluZXJDb25maWcScAoSa2ZwX3RyY
-  WluZXJfY29uZmlnGAIgASgLMikuc25hcGNoYXQucmVzZWFyY2guZ2JtbC5LRlBSZXNvdXJjZUNvbmZpZ0IV4j8SEhBrZnBUcmFpb
-  mVyQ29uZmlnSABSEGtmcFRyYWluZXJDb25maWcSeAoUbG9jYWxfdHJhaW5lcl9jb25maWcYAyABKAsyKy5zbmFwY2hhdC5yZXNlY
-  XJjaC5nYm1sLkxvY2FsUmVzb3VyY2VDb25maWdCF+I/FBISbG9jYWxUcmFpbmVyQ29uZmlnSABSEmxvY2FsVHJhaW5lckNvbmZpZ
-  0IQCg50cmFpbmVyX2NvbmZpZyLUAwoYSW5mZXJlbmNlclJlc291cmNlQ29uZmlnEo4BCht2ZXJ0ZXhfYWlfaW5mZXJlbmNlcl9jb
-  25maWcYASABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLlZlcnRleEFpUmVzb3VyY2VDb25maWdCHeI/GhIYdmVydGV4QWlJb
-  mZlcmVuY2VyQ29uZmlnSABSGHZlcnRleEFpSW5mZXJlbmNlckNvbmZpZxKNAQoaZGF0YWZsb3dfaW5mZXJlbmNlcl9jb25maWcYA
-  iABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRhdGFmbG93UmVzb3VyY2VDb25maWdCHeI/GhIYZGF0YWZsb3dJbmZlcmVuY
-  2VyQ29uZmlnSABSGGRhdGFmbG93SW5mZXJlbmNlckNvbmZpZxKBAQoXbG9jYWxfaW5mZXJlbmNlcl9jb25maWcYAyABKAsyKy5zb
-  mFwY2hhdC5yZXNlYXJjaC5nYm1sLkxvY2FsUmVzb3VyY2VDb25maWdCGuI/FxIVbG9jYWxJbmZlcmVuY2VyQ29uZmlnSABSFWxvY
-  2FsSW5mZXJlbmNlckNvbmZpZ0ITChFpbmZlcmVuY2VyX2NvbmZpZyKXCAoUU2hhcmVkUmVzb3VyY2VDb25maWcSfgoPcmVzb3VyY
-  2VfbGFiZWxzGAEgAygLMkAuc25hcGNoYXQucmVzZWFyY2guZ2JtbC5TaGFyZWRSZXNvdXJjZUNvbmZpZy5SZXNvdXJjZUxhYmVsc
-  0VudHJ5QhPiPxASDnJlc291cmNlTGFiZWxzUg5yZXNvdXJjZUxhYmVscxKOAQoVY29tbW9uX2NvbXB1dGVfY29uZmlnGAIgASgLM
-  kAuc25hcGNoYXQucmVzZWFyY2guZ2JtbC5TaGFyZWRSZXNvdXJjZUNvbmZpZy5Db21tb25Db21wdXRlQ29uZmlnQhjiPxUSE2Nvb
-  W1vbkNvbXB1dGVDb25maWdSE2NvbW1vbkNvbXB1dGVDb25maWcalAUKE0NvbW1vbkNvbXB1dGVDb25maWcSJgoHcHJvamVjdBgBI
-  AEoCUIM4j8JEgdwcm9qZWN0Ugdwcm9qZWN0EiMKBnJlZ2lvbhgCIAEoCUIL4j8IEgZyZWdpb25SBnJlZ2lvbhJDChJ0ZW1wX2Fzc
-  2V0c19idWNrZXQYAyABKAlCFeI/EhIQdGVtcEFzc2V0c0J1Y2tldFIQdGVtcEFzc2V0c0J1Y2tldBJcCht0ZW1wX3JlZ2lvbmFsX
-  2Fzc2V0c19idWNrZXQYBCABKAlCHeI/GhIYdGVtcFJlZ2lvbmFsQXNzZXRzQnVja2V0Uhh0ZW1wUmVnaW9uYWxBc3NldHNCdWNrZ
-  XQSQwoScGVybV9hc3NldHNfYnVja2V0GAUgASgJQhXiPxISEHBlcm1Bc3NldHNCdWNrZXRSEHBlcm1Bc3NldHNCdWNrZXQSWgobd
-  GVtcF9hc3NldHNfYnFfZGF0YXNldF9uYW1lGAYgASgJQhziPxkSF3RlbXBBc3NldHNCcURhdGFzZXROYW1lUhd0ZW1wQXNzZXRzQ
-  nFEYXRhc2V0TmFtZRJWChllbWJlZGRpbmdfYnFfZGF0YXNldF9uYW1lGAcgASgJQhviPxgSFmVtYmVkZGluZ0JxRGF0YXNldE5hb
-  WVSFmVtYmVkZGluZ0JxRGF0YXNldE5hbWUSVgoZZ2NwX3NlcnZpY2VfYWNjb3VudF9lbWFpbBgIIAEoCUIb4j8YEhZnY3BTZXJ2a
-  WNlQWNjb3VudEVtYWlsUhZnY3BTZXJ2aWNlQWNjb3VudEVtYWlsEjwKD2RhdGFmbG93X3J1bm5lchgLIAEoCUIT4j8QEg5kYXRhZ
-  mxvd1J1bm5lclIOZGF0YWZsb3dSdW5uZXIaVwoTUmVzb3VyY2VMYWJlbHNFbnRyeRIaCgNrZXkYASABKAlCCOI/BRIDa2V5UgNrZ
-  XkSIAoFdmFsdWUYAiABKAlCCuI/BxIFdmFsdWVSBXZhbHVlOgI4ASL3CAoSR2lnbFJlc291cmNlQ29uZmlnElsKGnNoYXJlZF9yZ
-  XNvdXJjZV9jb25maWdfdXJpGAEgASgJQhziPxkSF3NoYXJlZFJlc291cmNlQ29uZmlnVXJpSABSF3NoYXJlZFJlc291cmNlQ29uZ
-  mlnVXJpEn8KFnNoYXJlZF9yZXNvdXJjZV9jb25maWcYAiABKAsyLC5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLlNoYXJlZFJlc291c
-  mNlQ29uZmlnQhniPxYSFHNoYXJlZFJlc291cmNlQ29uZmlnSABSFHNoYXJlZFJlc291cmNlQ29uZmlnEngKE3ByZXByb2Nlc3Nvc
-  l9jb25maWcYDCABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRhdGFQcmVwcm9jZXNzb3JDb25maWdCF+I/FBIScHJlcHJvY
-  2Vzc29yQ29uZmlnUhJwcmVwcm9jZXNzb3JDb25maWcSfwoXc3ViZ3JhcGhfc2FtcGxlcl9jb25maWcYDSABKAsyKy5zbmFwY2hhd
-  C5yZXNlYXJjaC5nYm1sLlNwYXJrUmVzb3VyY2VDb25maWdCGuI/FxIVc3ViZ3JhcGhTYW1wbGVyQ29uZmlnUhVzdWJncmFwaFNhb
-  XBsZXJDb25maWcSfAoWc3BsaXRfZ2VuZXJhdG9yX2NvbmZpZxgOIAEoCzIrLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuU3BhcmtSZ
-  XNvdXJjZUNvbmZpZ0IZ4j8WEhRzcGxpdEdlbmVyYXRvckNvbmZpZ1IUc3BsaXRHZW5lcmF0b3JDb25maWcSbQoOdHJhaW5lcl9jb
-  25maWcYDyABKAsyMC5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRpc3RyaWJ1dGVkVHJhaW5lckNvbmZpZ0IUGAHiPw8SDXRyYWluZ
-  XJDb25maWdSDXRyYWluZXJDb25maWcSdAoRaW5mZXJlbmNlcl9jb25maWcYECABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sL
-  kRhdGFmbG93UmVzb3VyY2VDb25maWdCFxgB4j8SEhBpbmZlcmVuY2VyQ29uZmlnUhBpbmZlcmVuY2VyQ29uZmlnEoEBChd0cmFpb
-  mVyX3Jlc291cmNlX2NvbmZpZxgRIAEoCzItLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuVHJhaW5lclJlc291cmNlQ29uZmlnQhriP
-  xcSFXRyYWluZXJSZXNvdXJjZUNvbmZpZ1IVdHJhaW5lclJlc291cmNlQ29uZmlnEo0BChppbmZlcmVuY2VyX3Jlc291cmNlX2Nvb
-  mZpZxgSIAEoCzIwLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuSW5mZXJlbmNlclJlc291cmNlQ29uZmlnQh3iPxoSGGluZmVyZW5jZ
-  XJSZXNvdXJjZUNvbmZpZ1IYaW5mZXJlbmNlclJlc291cmNlQ29uZmlnQhEKD3NoYXJlZF9yZXNvdXJjZSrjAwoJQ29tcG9uZW50E
-  i0KEUNvbXBvbmVudF9Vbmtub3duEAAaFuI/ExIRQ29tcG9uZW50X1Vua25vd24SPwoaQ29tcG9uZW50X0NvbmZpZ19WYWxpZGF0b
-  3IQARof4j8cEhpDb21wb25lbnRfQ29uZmlnX1ZhbGlkYXRvchI/ChpDb21wb25lbnRfQ29uZmlnX1BvcHVsYXRvchACGh/iPxwSG
-  kNvbXBvbmVudF9Db25maWdfUG9wdWxhdG9yEkEKG0NvbXBvbmVudF9EYXRhX1ByZXByb2Nlc3NvchADGiDiPx0SG0NvbXBvbmVud
-  F9EYXRhX1ByZXByb2Nlc3NvchI/ChpDb21wb25lbnRfU3ViZ3JhcGhfU2FtcGxlchAEGh/iPxwSGkNvbXBvbmVudF9TdWJncmFwa
-  F9TYW1wbGVyEj0KGUNvbXBvbmVudF9TcGxpdF9HZW5lcmF0b3IQBRoe4j8bEhlDb21wb25lbnRfU3BsaXRfR2VuZXJhdG9yEi0KE
-  UNvbXBvbmVudF9UcmFpbmVyEAYaFuI/ExIRQ29tcG9uZW50X1RyYWluZXISMwoUQ29tcG9uZW50X0luZmVyZW5jZXIQBxoZ4j8WE
-  hRDb21wb25lbnRfSW5mZXJlbmNlcmIGcHJvdG8z"""
+  GljYXMSJgoHdGltZW91dBgFIAEoDUIM4j8JEgd0aW1lb3V0Ugd0aW1lb3V0EkYKE2djcF9yZWdpb25fb3ZlcnJpZGUYBiABKAlCF
+  uI/ExIRZ2NwUmVnaW9uT3ZlcnJpZGVSEWdjcFJlZ2lvbk92ZXJyaWRlIooCChFLRlBSZXNvdXJjZUNvbmZpZxIwCgtjcHVfcmVxd
+  WVzdBgBIAEoCUIP4j8MEgpjcHVSZXF1ZXN0UgpjcHVSZXF1ZXN0EjkKDm1lbW9yeV9yZXF1ZXN0GAIgASgJQhLiPw8SDW1lbW9ye
+  VJlcXVlc3RSDW1lbW9yeVJlcXVlc3QSJwoIZ3B1X3R5cGUYAyABKAlCDOI/CRIHZ3B1VHlwZVIHZ3B1VHlwZRIqCglncHVfbGlta
+  XQYBCABKA1CDeI/ChIIZ3B1TGltaXRSCGdwdUxpbWl0EjMKDG51bV9yZXBsaWNhcxgFIAEoDUIQ4j8NEgtudW1SZXBsaWNhc1ILb
+  nVtUmVwbGljYXMiRwoTTG9jYWxSZXNvdXJjZUNvbmZpZxIwCgtudW1fd29ya2VycxgBIAEoDUIP4j8MEgpudW1Xb3JrZXJzUgpud
+  W1Xb3JrZXJzIp0DChhEaXN0cmlidXRlZFRyYWluZXJDb25maWcShAEKGHZlcnRleF9haV90cmFpbmVyX2NvbmZpZxgBIAEoCzItL
+  nNuYXBjaGF0LnJlc2VhcmNoLmdibWwuVmVydGV4QWlUcmFpbmVyQ29uZmlnQhriPxcSFXZlcnRleEFpVHJhaW5lckNvbmZpZ0gAU
+  hV2ZXJ0ZXhBaVRyYWluZXJDb25maWcSbwoSa2ZwX3RyYWluZXJfY29uZmlnGAIgASgLMiguc25hcGNoYXQucmVzZWFyY2guZ2Jtb
+  C5LRlBUcmFpbmVyQ29uZmlnQhXiPxISEGtmcFRyYWluZXJDb25maWdIAFIQa2ZwVHJhaW5lckNvbmZpZxJ3ChRsb2NhbF90cmFpb
+  mVyX2NvbmZpZxgDIAEoCzIqLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuTG9jYWxUcmFpbmVyQ29uZmlnQhfiPxQSEmxvY2FsVHJha
+  W5lckNvbmZpZ0gAUhJsb2NhbFRyYWluZXJDb25maWdCEAoOdHJhaW5lcl9jb25maWcinQMKFVRyYWluZXJSZXNvdXJjZUNvbmZpZ
+  xKFAQoYdmVydGV4X2FpX3RyYWluZXJfY29uZmlnGAEgASgLMi4uc25hcGNoYXQucmVzZWFyY2guZ2JtbC5WZXJ0ZXhBaVJlc291c
+  mNlQ29uZmlnQhriPxcSFXZlcnRleEFpVHJhaW5lckNvbmZpZ0gAUhV2ZXJ0ZXhBaVRyYWluZXJDb25maWcScAoSa2ZwX3RyYWluZ
+  XJfY29uZmlnGAIgASgLMikuc25hcGNoYXQucmVzZWFyY2guZ2JtbC5LRlBSZXNvdXJjZUNvbmZpZ0IV4j8SEhBrZnBUcmFpbmVyQ
+  29uZmlnSABSEGtmcFRyYWluZXJDb25maWcSeAoUbG9jYWxfdHJhaW5lcl9jb25maWcYAyABKAsyKy5zbmFwY2hhdC5yZXNlYXJja
+  C5nYm1sLkxvY2FsUmVzb3VyY2VDb25maWdCF+I/FBISbG9jYWxUcmFpbmVyQ29uZmlnSABSEmxvY2FsVHJhaW5lckNvbmZpZ0IQC
+  g50cmFpbmVyX2NvbmZpZyLUAwoYSW5mZXJlbmNlclJlc291cmNlQ29uZmlnEo4BCht2ZXJ0ZXhfYWlfaW5mZXJlbmNlcl9jb25ma
+  WcYASABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLlZlcnRleEFpUmVzb3VyY2VDb25maWdCHeI/GhIYdmVydGV4QWlJbmZlc
+  mVuY2VyQ29uZmlnSABSGHZlcnRleEFpSW5mZXJlbmNlckNvbmZpZxKNAQoaZGF0YWZsb3dfaW5mZXJlbmNlcl9jb25maWcYAiABK
+  AsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRhdGFmbG93UmVzb3VyY2VDb25maWdCHeI/GhIYZGF0YWZsb3dJbmZlcmVuY2VyQ
+  29uZmlnSABSGGRhdGFmbG93SW5mZXJlbmNlckNvbmZpZxKBAQoXbG9jYWxfaW5mZXJlbmNlcl9jb25maWcYAyABKAsyKy5zbmFwY
+  2hhdC5yZXNlYXJjaC5nYm1sLkxvY2FsUmVzb3VyY2VDb25maWdCGuI/FxIVbG9jYWxJbmZlcmVuY2VyQ29uZmlnSABSFWxvY2FsS
+  W5mZXJlbmNlckNvbmZpZ0ITChFpbmZlcmVuY2VyX2NvbmZpZyKXCAoUU2hhcmVkUmVzb3VyY2VDb25maWcSfgoPcmVzb3VyY2Vfb
+  GFiZWxzGAEgAygLMkAuc25hcGNoYXQucmVzZWFyY2guZ2JtbC5TaGFyZWRSZXNvdXJjZUNvbmZpZy5SZXNvdXJjZUxhYmVsc0Vud
+  HJ5QhPiPxASDnJlc291cmNlTGFiZWxzUg5yZXNvdXJjZUxhYmVscxKOAQoVY29tbW9uX2NvbXB1dGVfY29uZmlnGAIgASgLMkAuc
+  25hcGNoYXQucmVzZWFyY2guZ2JtbC5TaGFyZWRSZXNvdXJjZUNvbmZpZy5Db21tb25Db21wdXRlQ29uZmlnQhjiPxUSE2NvbW1vb
+  kNvbXB1dGVDb25maWdSE2NvbW1vbkNvbXB1dGVDb25maWcalAUKE0NvbW1vbkNvbXB1dGVDb25maWcSJgoHcHJvamVjdBgBIAEoC
+  UIM4j8JEgdwcm9qZWN0Ugdwcm9qZWN0EiMKBnJlZ2lvbhgCIAEoCUIL4j8IEgZyZWdpb25SBnJlZ2lvbhJDChJ0ZW1wX2Fzc2V0c
+  19idWNrZXQYAyABKAlCFeI/EhIQdGVtcEFzc2V0c0J1Y2tldFIQdGVtcEFzc2V0c0J1Y2tldBJcCht0ZW1wX3JlZ2lvbmFsX2Fzc
+  2V0c19idWNrZXQYBCABKAlCHeI/GhIYdGVtcFJlZ2lvbmFsQXNzZXRzQnVja2V0Uhh0ZW1wUmVnaW9uYWxBc3NldHNCdWNrZXQSQ
+  woScGVybV9hc3NldHNfYnVja2V0GAUgASgJQhXiPxISEHBlcm1Bc3NldHNCdWNrZXRSEHBlcm1Bc3NldHNCdWNrZXQSWgobdGVtc
+  F9hc3NldHNfYnFfZGF0YXNldF9uYW1lGAYgASgJQhziPxkSF3RlbXBBc3NldHNCcURhdGFzZXROYW1lUhd0ZW1wQXNzZXRzQnFEY
+  XRhc2V0TmFtZRJWChllbWJlZGRpbmdfYnFfZGF0YXNldF9uYW1lGAcgASgJQhviPxgSFmVtYmVkZGluZ0JxRGF0YXNldE5hbWVSF
+  mVtYmVkZGluZ0JxRGF0YXNldE5hbWUSVgoZZ2NwX3NlcnZpY2VfYWNjb3VudF9lbWFpbBgIIAEoCUIb4j8YEhZnY3BTZXJ2aWNlQ
+  WNjb3VudEVtYWlsUhZnY3BTZXJ2aWNlQWNjb3VudEVtYWlsEjwKD2RhdGFmbG93X3J1bm5lchgLIAEoCUIT4j8QEg5kYXRhZmxvd
+  1J1bm5lclIOZGF0YWZsb3dSdW5uZXIaVwoTUmVzb3VyY2VMYWJlbHNFbnRyeRIaCgNrZXkYASABKAlCCOI/BRIDa2V5UgNrZXkSI
+  AoFdmFsdWUYAiABKAlCCuI/BxIFdmFsdWVSBXZhbHVlOgI4ASL3CAoSR2lnbFJlc291cmNlQ29uZmlnElsKGnNoYXJlZF9yZXNvd
+  XJjZV9jb25maWdfdXJpGAEgASgJQhziPxkSF3NoYXJlZFJlc291cmNlQ29uZmlnVXJpSABSF3NoYXJlZFJlc291cmNlQ29uZmlnV
+  XJpEn8KFnNoYXJlZF9yZXNvdXJjZV9jb25maWcYAiABKAsyLC5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLlNoYXJlZFJlc291cmNlQ
+  29uZmlnQhniPxYSFHNoYXJlZFJlc291cmNlQ29uZmlnSABSFHNoYXJlZFJlc291cmNlQ29uZmlnEngKE3ByZXByb2Nlc3Nvcl9jb
+  25maWcYDCABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRhdGFQcmVwcm9jZXNzb3JDb25maWdCF+I/FBIScHJlcHJvY2Vzc
+  29yQ29uZmlnUhJwcmVwcm9jZXNzb3JDb25maWcSfwoXc3ViZ3JhcGhfc2FtcGxlcl9jb25maWcYDSABKAsyKy5zbmFwY2hhdC5yZ
+  XNlYXJjaC5nYm1sLlNwYXJrUmVzb3VyY2VDb25maWdCGuI/FxIVc3ViZ3JhcGhTYW1wbGVyQ29uZmlnUhVzdWJncmFwaFNhbXBsZ
+  XJDb25maWcSfAoWc3BsaXRfZ2VuZXJhdG9yX2NvbmZpZxgOIAEoCzIrLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuU3BhcmtSZXNvd
+  XJjZUNvbmZpZ0IZ4j8WEhRzcGxpdEdlbmVyYXRvckNvbmZpZ1IUc3BsaXRHZW5lcmF0b3JDb25maWcSbQoOdHJhaW5lcl9jb25ma
+  WcYDyABKAsyMC5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRpc3RyaWJ1dGVkVHJhaW5lckNvbmZpZ0IUGAHiPw8SDXRyYWluZXJDb
+  25maWdSDXRyYWluZXJDb25maWcSdAoRaW5mZXJlbmNlcl9jb25maWcYECABKAsyLi5zbmFwY2hhdC5yZXNlYXJjaC5nYm1sLkRhd
+  GFmbG93UmVzb3VyY2VDb25maWdCFxgB4j8SEhBpbmZlcmVuY2VyQ29uZmlnUhBpbmZlcmVuY2VyQ29uZmlnEoEBChd0cmFpbmVyX
+  3Jlc291cmNlX2NvbmZpZxgRIAEoCzItLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuVHJhaW5lclJlc291cmNlQ29uZmlnQhriPxcSF
+  XRyYWluZXJSZXNvdXJjZUNvbmZpZ1IVdHJhaW5lclJlc291cmNlQ29uZmlnEo0BChppbmZlcmVuY2VyX3Jlc291cmNlX2NvbmZpZ
+  xgSIAEoCzIwLnNuYXBjaGF0LnJlc2VhcmNoLmdibWwuSW5mZXJlbmNlclJlc291cmNlQ29uZmlnQh3iPxoSGGluZmVyZW5jZXJSZ
+  XNvdXJjZUNvbmZpZ1IYaW5mZXJlbmNlclJlc291cmNlQ29uZmlnQhEKD3NoYXJlZF9yZXNvdXJjZSrjAwoJQ29tcG9uZW50Ei0KE
+  UNvbXBvbmVudF9Vbmtub3duEAAaFuI/ExIRQ29tcG9uZW50X1Vua25vd24SPwoaQ29tcG9uZW50X0NvbmZpZ19WYWxpZGF0b3IQA
+  Rof4j8cEhpDb21wb25lbnRfQ29uZmlnX1ZhbGlkYXRvchI/ChpDb21wb25lbnRfQ29uZmlnX1BvcHVsYXRvchACGh/iPxwSGkNvb
+  XBvbmVudF9Db25maWdfUG9wdWxhdG9yEkEKG0NvbXBvbmVudF9EYXRhX1ByZXByb2Nlc3NvchADGiDiPx0SG0NvbXBvbmVudF9EY
+  XRhX1ByZXByb2Nlc3NvchI/ChpDb21wb25lbnRfU3ViZ3JhcGhfU2FtcGxlchAEGh/iPxwSGkNvbXBvbmVudF9TdWJncmFwaF9TY
+  W1wbGVyEj0KGUNvbXBvbmVudF9TcGxpdF9HZW5lcmF0b3IQBRoe4j8bEhlDb21wb25lbnRfU3BsaXRfR2VuZXJhdG9yEi0KEUNvb
+  XBvbmVudF9UcmFpbmVyEAYaFuI/ExIRQ29tcG9uZW50X1RyYWluZXISMwoUQ29tcG9uZW50X0luZmVyZW5jZXIQBxoZ4j8WEhRDb
+  21wb25lbnRfSW5mZXJlbmNlcmIGcHJvdG8z"""
       ).mkString)
   lazy val scalaDescriptor: _root_.scalapb.descriptors.FileDescriptor = {
     val scalaProto = com.google.protobuf.descriptor.FileDescriptorProto.parseFrom(ProtoBytes)
@@ -118,4 +119,4 @@ object GiglResourceConfigProto extends _root_.scalapb.GeneratedFileObject {
   }
   @deprecated("Use javaDescriptor instead. In a future version this will refer to scalaDescriptor.", "ScalaPB 0.5.47")
   def descriptor: com.google.protobuf.Descriptors.FileDescriptor = javaDescriptor
-}
\ No newline at end of file
+}
diff --git a/scala_spark35/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/VertexAiResourceConfig.scala b/scala_spark35/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/VertexAiResourceConfig.scala
index fbbdb297d..152f96abb 100644
--- a/scala_spark35/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/VertexAiResourceConfig.scala
+++ b/scala_spark35/common/src/main/scala/snapchat/research/gbml/gigl_resource_config/VertexAiResourceConfig.scala
@@ -16,8 +16,13 @@ package snapchat.research.gbml.gigl_resource_config
   * @param numReplicas
   *   Num workers for job
   * @param timeout
-  *   Timeout in seconds for the job. If unset or zero, will use the default @ google.cloud.aiplatform.CustomJob, which is 7 days: 
+  *   Timeout in seconds for the job. If unset or zero, will use the default @ google.cloud.aiplatform.CustomJob, which is 7 days:
   *   https://github.com/googleapis/python-aiplatform/blob/58fbabdeeefd1ccf1a9d0c22eeb5606aeb9c2266/google/cloud/aiplatform/jobs.py#L2252-L2253
+  * @param gcpRegionOverride
+  *   Region override.
+  *   If provided, then the Vertex AI Job will be launched in the provided region.
+  *   ex: "us-west1"
+  *   NOTE: If set, then there may be data egress costs from CommonComputeConfig.region -> gcp_region_override
   */
 @SerialVersionUID(0L)
 final case class VertexAiResourceConfig(
@@ -26,47 +31,55 @@ final case class VertexAiResourceConfig(
     gpuLimit: _root_.scala.Int = 0,
     numReplicas: _root_.scala.Int = 0,
     timeout: _root_.scala.Int = 0,
+    gcpRegionOverride: _root_.scala.Predef.String = "",
     unknownFields: _root_.scalapb.UnknownFieldSet = _root_.scalapb.UnknownFieldSet.empty
     ) extends scalapb.GeneratedMessage with scalapb.lenses.Updatable[VertexAiResourceConfig] {
     @transient
     private[this] var __serializedSizeMemoized: _root_.scala.Int = 0
     private[this] def __computeSerializedSize(): _root_.scala.Int = {
       var __size = 0
-      
+
       {
         val __value = machineType
         if (!__value.isEmpty) {
           __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(1, __value)
         }
       };
-      
+
       {
         val __value = gpuType
         if (!__value.isEmpty) {
           __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(2, __value)
         }
       };
-      
+
       {
         val __value = gpuLimit
         if (__value != 0) {
           __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(3, __value)
         }
       };
-      
+
       {
         val __value = numReplicas
         if (__value != 0) {
           __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(4, __value)
         }
       };
-      
+
       {
         val __value = timeout
         if (__value != 0) {
           __size += _root_.com.google.protobuf.CodedOutputStream.computeUInt32Size(5, __value)
         }
       };
+
+      {
+        val __value = gcpRegionOverride
+        if (!__value.isEmpty) {
+          __size += _root_.com.google.protobuf.CodedOutputStream.computeStringSize(6, __value)
+        }
+      };
       __size += unknownFields.serializedSize
       __size
     }
@@ -77,7 +90,7 @@ final case class VertexAiResourceConfig(
         __serializedSizeMemoized = __size
       }
       __size - 1
-      
+
     }
     def writeTo(`_output__`: _root_.com.google.protobuf.CodedOutputStream): _root_.scala.Unit = {
       {
@@ -110,6 +123,12 @@ final case class VertexAiResourceConfig(
           _output__.writeUInt32(5, __v)
         }
       };
+      {
+        val __v = gcpRegionOverride
+        if (!__v.isEmpty) {
+          _output__.writeString(6, __v)
+        }
+      };
       unknownFields.writeTo(_output__)
     }
     def withMachineType(__v: _root_.scala.Predef.String): VertexAiResourceConfig = copy(machineType = __v)
@@ -117,6 +136,7 @@ final case class VertexAiResourceConfig(
     def withGpuLimit(__v: _root_.scala.Int): VertexAiResourceConfig = copy(gpuLimit = __v)
     def withNumReplicas(__v: _root_.scala.Int): VertexAiResourceConfig = copy(numReplicas = __v)
     def withTimeout(__v: _root_.scala.Int): VertexAiResourceConfig = copy(timeout = __v)
+    def withGcpRegionOverride(__v: _root_.scala.Predef.String): VertexAiResourceConfig = copy(gcpRegionOverride = __v)
     def withUnknownFields(__v: _root_.scalapb.UnknownFieldSet) = copy(unknownFields = __v)
     def discardUnknownFields = copy(unknownFields = _root_.scalapb.UnknownFieldSet.empty)
     def getFieldByNumber(__fieldNumber: _root_.scala.Int): _root_.scala.Any = {
@@ -141,6 +161,10 @@ final case class VertexAiResourceConfig(
           val __t = timeout
           if (__t != 0) __t else null
         }
+        case 6 => {
+          val __t = gcpRegionOverride
+          if (__t != "") __t else null
+        }
       }
     }
     def getField(__field: _root_.scalapb.descriptors.FieldDescriptor): _root_.scalapb.descriptors.PValue = {
@@ -151,6 +175,7 @@ final case class VertexAiResourceConfig(
         case 3 => _root_.scalapb.descriptors.PInt(gpuLimit)
         case 4 => _root_.scalapb.descriptors.PInt(numReplicas)
         case 5 => _root_.scalapb.descriptors.PInt(timeout)
+        case 6 => _root_.scalapb.descriptors.PString(gcpRegionOverride)
       }
     }
     def toProtoString: _root_.scala.Predef.String = _root_.scalapb.TextFormat.printToUnicodeString(this)
@@ -166,6 +191,7 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
     var __gpuLimit: _root_.scala.Int = 0
     var __numReplicas: _root_.scala.Int = 0
     var __timeout: _root_.scala.Int = 0
+    var __gcpRegionOverride: _root_.scala.Predef.String = ""
     var `_unknownFields__`: _root_.scalapb.UnknownFieldSet.Builder = null
     var _done__ = false
     while (!_done__) {
@@ -182,6 +208,8 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
           __numReplicas = _input__.readUInt32()
         case 40 =>
           __timeout = _input__.readUInt32()
+        case 50 =>
+          __gcpRegionOverride = _input__.readStringRequireUtf8()
         case tag =>
           if (_unknownFields__ == null) {
             _unknownFields__ = new _root_.scalapb.UnknownFieldSet.Builder()
@@ -195,6 +223,7 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
         gpuLimit = __gpuLimit,
         numReplicas = __numReplicas,
         timeout = __timeout,
+        gcpRegionOverride = __gcpRegionOverride,
         unknownFields = if (_unknownFields__ == null) _root_.scalapb.UnknownFieldSet.empty else _unknownFields__.result()
     )
   }
@@ -206,7 +235,8 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
         gpuType = __fieldsMap.get(scalaDescriptor.findFieldByNumber(2).get).map(_.as[_root_.scala.Predef.String]).getOrElse(""),
         gpuLimit = __fieldsMap.get(scalaDescriptor.findFieldByNumber(3).get).map(_.as[_root_.scala.Int]).getOrElse(0),
         numReplicas = __fieldsMap.get(scalaDescriptor.findFieldByNumber(4).get).map(_.as[_root_.scala.Int]).getOrElse(0),
-        timeout = __fieldsMap.get(scalaDescriptor.findFieldByNumber(5).get).map(_.as[_root_.scala.Int]).getOrElse(0)
+        timeout = __fieldsMap.get(scalaDescriptor.findFieldByNumber(5).get).map(_.as[_root_.scala.Int]).getOrElse(0),
+        gcpRegionOverride = __fieldsMap.get(scalaDescriptor.findFieldByNumber(6).get).map(_.as[_root_.scala.Predef.String]).getOrElse("")
       )
     case _ => throw new RuntimeException("Expected PMessage")
   }
@@ -220,7 +250,8 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
     gpuType = "",
     gpuLimit = 0,
     numReplicas = 0,
-    timeout = 0
+    timeout = 0,
+    gcpRegionOverride = ""
   )
   implicit class VertexAiResourceConfigLens[UpperPB](_l: _root_.scalapb.lenses.Lens[UpperPB, snapchat.research.gbml.gigl_resource_config.VertexAiResourceConfig]) extends _root_.scalapb.lenses.ObjectLens[UpperPB, snapchat.research.gbml.gigl_resource_config.VertexAiResourceConfig](_l) {
     def machineType: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.machineType)((c_, f_) => c_.copy(machineType = f_))
@@ -228,24 +259,28 @@ object VertexAiResourceConfig extends scalapb.GeneratedMessageCompanion[snapchat
     def gpuLimit: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Int] = field(_.gpuLimit)((c_, f_) => c_.copy(gpuLimit = f_))
     def numReplicas: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Int] = field(_.numReplicas)((c_, f_) => c_.copy(numReplicas = f_))
     def timeout: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Int] = field(_.timeout)((c_, f_) => c_.copy(timeout = f_))
+    def gcpRegionOverride: _root_.scalapb.lenses.Lens[UpperPB, _root_.scala.Predef.String] = field(_.gcpRegionOverride)((c_, f_) => c_.copy(gcpRegionOverride = f_))
   }
   final val MACHINE_TYPE_FIELD_NUMBER = 1
   final val GPU_TYPE_FIELD_NUMBER = 2
   final val GPU_LIMIT_FIELD_NUMBER = 3
   final val NUM_REPLICAS_FIELD_NUMBER = 4
   final val TIMEOUT_FIELD_NUMBER = 5
+  final val GCP_REGION_OVERRIDE_FIELD_NUMBER = 6
   def of(
     machineType: _root_.scala.Predef.String,
     gpuType: _root_.scala.Predef.String,
     gpuLimit: _root_.scala.Int,
     numReplicas: _root_.scala.Int,
-    timeout: _root_.scala.Int
+    timeout: _root_.scala.Int,
+    gcpRegionOverride: _root_.scala.Predef.String
   ): _root_.snapchat.research.gbml.gigl_resource_config.VertexAiResourceConfig = _root_.snapchat.research.gbml.gigl_resource_config.VertexAiResourceConfig(
     machineType,
     gpuType,
     gpuLimit,
     numReplicas,
-    timeout
+    timeout,
+    gcpRegionOverride
   )
   // @@protoc_insertion_point(GeneratedMessageCompanion[snapchat.research.gbml.VertexAiResourceConfig])
 }