code review comments

deepanker13 · Jan 4, 2024 · b8caae1 · b8caae1
1 parent cd811cd
commit b8caae1
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/sdk/python/kubeflow/training/api/training_client.py b/sdk/python/kubeflow/training/api/training_client.py
@@ -32,7 +32,6 @@
     HfDatasetParams,
     INIT_CONTAINER_MOUNT_PATH,
 )
-import os
 
 logger = logging.getLogger(__name__)
 
@@ -101,10 +100,7 @@ def train(
         model_provider_parameters: HuggingFaceModelParams = None,
         dataset_provider_parameters: Union[HfDatasetParams, S3DatasetParams] = None,
         train_parameters: HuggingFaceTrainParams = None,
-        resources_per_worker: Union[dict, client.V1ResourceRequirements, None] = {
-            "cpu": 1,
-            "memory": "2Gi",
-        },
+        resources_per_worker: Union[dict, client.V1ResourceRequirements, None] = None,
         # Dict[Literal["gpu", "cpu", "memory"], any] = None,
     ):
         """