Skip to content

Commit f6baa76

Browse files
committed
Format code and revert toml
1 parent b6c596c commit f6baa76

File tree

2 files changed

+15
-15
lines changed

2 files changed

+15
-15
lines changed

dlc_developer_config.toml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ deep_canary_mode = false
3737
[build]
3838
# Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image.
3939
# available frameworks - ["base", "vllm", "autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
40-
build_frameworks = ["base", "pytorch"]
40+
build_frameworks = []
4141

4242

4343
# By default we build both training and inference containers. Set true/false values to determine which to build.
@@ -58,12 +58,12 @@ notify_test_failures = false
5858
[test]
5959
### On by default
6060
sanity_tests = true
61-
security_tests = false
61+
security_tests = true
6262
safety_check_test = false
6363
ecr_scan_allowlist_feature = false
64-
ecs_tests = false
65-
eks_tests = false
66-
ec2_tests = false
64+
ecs_tests = true
65+
eks_tests = true
66+
ec2_tests = true
6767
# Set it to true if you are preparing a Benchmark related PR
6868
ec2_benchmark_tests = false
6969

@@ -74,7 +74,7 @@ ec2_benchmark_tests = false
7474
ec2_tests_on_heavy_instances = false
7575
### SM specific tests
7676
### On by default
77-
sagemaker_local_tests = false
77+
sagemaker_local_tests = true
7878
### Set enable_ipv6 = true to run tests with IPv6-enabled resources
7979
### Off by default (set to false)
8080
enable_ipv6 = false
@@ -92,7 +92,7 @@ enable_ipv6 = false
9292
ipv6_vpc_name = ""
9393

9494
# run standard sagemaker remote tests from test/sagemaker_tests
95-
sagemaker_remote_tests = false
95+
sagemaker_remote_tests = true
9696
# run efa sagemaker tests
9797
sagemaker_efa_tests = false
9898
# run release_candidate_integration tests

src/image.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ def build(self):
200200
def docker_build(self, context_path, custom_context=False):
201201
"""
202202
Uses Docker Buildx for vLLM images, falls back to legacy Docker API for others
203-
203+
204204
:param context_path: str, Path to build context
205205
:param custom_context: bool, Whether to use custom context from stdin (default: False)
206206
:return: int, Build status
@@ -215,13 +215,13 @@ def docker_build(self, context_path, custom_context=False):
215215
def _is_vllm_image(self):
216216
"""
217217
Determine if current image is a vLLM image
218-
218+
219219
:return: bool, True if this is a vLLM image
220220
"""
221221
return (
222-
self.info.get("framework") == "vllm" or
223-
"vllm" in self.repository.lower() or
224-
"vllm" in str(self.info.get("name", "")).lower()
222+
self.info.get("framework") == "vllm"
223+
or "vllm" in self.repository.lower()
224+
or "vllm" in str(self.info.get("name", "")).lower()
225225
)
226226

227227
def _buildx_build(self, context_path, custom_context=False):
@@ -318,7 +318,7 @@ def _buildx_build(self, context_path, custom_context=False):
318318
def _legacy_docker_build(self, context_path, custom_context=False):
319319
"""
320320
Uses legacy Docker API Client to build the image (for non-vLLM images).
321-
321+
322322
:param context_path: str, Path to build context
323323
:param custom_context: bool, Whether to use custom context from stdin (default: False)
324324
:return: int, Build Status
@@ -331,7 +331,7 @@ def _legacy_docker_build(self, context_path, custom_context=False):
331331

332332
line_counter = 0
333333
line_interval = 50
334-
334+
335335
try:
336336
for line in self.client.build(
337337
fileobj=fileobj,
@@ -378,7 +378,7 @@ def _legacy_docker_build(self, context_path, custom_context=False):
378378

379379
self.build_status = constants.SUCCESS
380380
return self.build_status
381-
381+
382382
except Exception as e:
383383
response.append(f"Legacy Docker build error: {str(e)}")
384384
self.build_status = constants.FAIL

0 commit comments

Comments
 (0)