Merge remote-tracking branch 'upstream/main'

Author: dchourasia

.pre-commit-config.yaml

@@ -0,0 +1,30 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v2.3.0
+    hooks:
+      - id: check-yaml
+        args: [--allow-multiple-documents]
+      - id: check-json
+      - id: end-of-file-fixer
+      - id: trailing-whitespace
+      - id: pretty-format-json
+
+  - repo: https://github.com/pycqa/isort
+    rev: 5.11.5
+    hooks:
+      - id: isort
+        name: isort
+        entry: isort --profile black
+
+  - repo: https://github.com/psf/black
+    rev: 24.2.0
+    hooks:
+      - id: black
+
+  - repo: https://github.com/pycqa/flake8
+    rev: 7.1.1
+    hooks:
+      - id: flake8
+        args:
+          - --ignore=E203,W503
+          - --max-line-length=88

Makefile

@@ -45,3 +45,7 @@ build-test-image:
 .PHONY: push-test-image
 push-test-image:
 	podman push ${E2E_TEST_IMAGE}
+
+.PHONY: precommit
+precommit:
+	pre-commit run --all-files

tests/kfto/kfto_training_test.go

@@ -341,7 +341,7 @@ func createKFTOPyTorchJob(test Test, namespace string, config corev1.ConfigMap,
 										},
 										Limits: corev1.ResourceList{
 											corev1.ResourceCPU:                     resource.MustParse("2"),
-											corev1.ResourceMemory:                  resource.MustParse("8Gi"),
+											corev1.ResourceMemory:                  resource.MustParse("12Gi"),
 											corev1.ResourceName(gpu.ResourceLabel): resource.MustParse(fmt.Sprint(numGpus)),
 										},
 									},
@@ -506,7 +506,7 @@ func createKFTOPyTorchJob(test Test, namespace string, config corev1.ConfigMap,
 								},
 								Limits: corev1.ResourceList{
 									corev1.ResourceCPU:                     resource.MustParse("2"),
-									corev1.ResourceMemory:                  resource.MustParse("8Gi"),
+									corev1.ResourceMemory:                  resource.MustParse("12Gi"),
 									corev1.ResourceName(gpu.ResourceLabel): resource.MustParse(fmt.Sprint(numGpus)),
 								},
 							},