File tree Expand file tree Collapse file tree 1 file changed +0
-38
lines changed
src/automation/tasks/scripts Expand file tree Collapse file tree 1 file changed +0
-38
lines changed Original file line number Diff line number Diff line change 5
5
from automation .vllm import start_vllm_server
6
6
from pyhocon import ConfigFactory
7
7
8
- """
9
- def main(configurations=None):
10
- task = Task.current_task()
11
-
12
- args = task.get_parameters_as_dict(cast=True)
13
-
14
- raw_config = task.get_configuration_object("GuideLLM")
15
- if raw_config is None:
16
- print("[DEBUG] `GuideLLM` config not found in configuration — checking parameters as fallback")
17
- raw_config = task.get_parameters_as_dict().get("GuideLLM")
18
- if raw_config is None:
19
- raise RuntimeError("GuideLLM config is None. This likely means `get_configurations()` is not returning it or it's not passed via parameters.")
20
- guidellm_args = ConfigFactory.from_dict(raw_config)
21
- else:
22
- guidellm_args = ConfigFactory.parse_string(raw_config)
23
-
24
- def clean_hocon_value(v):
25
- if isinstance(v, str) and v.startswith('"') and v.endswith('"'):
26
- return v[1:-1]
27
- return v
28
-
29
- guidellm_args = {k: clean_hocon_value(v) for k, v in guidellm_args.items()}
30
-
31
- print("[DEBUG] Guidellm_Args:", guidellm_args)
32
-
33
- environment_args = task.get_configuration_object("environment")
34
- if environment_args is None:
35
- environment_args = {}
36
- else:
37
- environment_args = ConfigFactory.parse_string(environment_args)
38
-
39
- vllm_args = task.get_configuration_object("vLLM")
40
- if vllm_args is None:
41
- vllm_args = {}
42
- else:
43
- vllm_args = ConfigFactory.parse_string(vllm_args)
44
- """
45
-
46
8
def main (configurations = None ):
47
9
task = Task .current_task ()
48
10
You can’t perform that action at this time.
0 commit comments