ATLAS Offline Software
Loading...
Searching...
No Matches
python.TritonToolConfig Namespace Reference

Functions

 TritonToolCfg (flags, str model_name, str url, int port=8001, str model_version="", float timeout=0., bool ssl=False, name="TritonTool", **kwargs)

Function Documentation

◆ TritonToolCfg()

python.TritonToolConfig.TritonToolCfg ( flags,
str model_name,
str url,
int port = 8001,
str model_version = "",
float timeout = 0.,
bool ssl = False,
name = "TritonTool",
** kwargs )
Configure TritonTool in Control/AthOnnx/AthTritonComps/src

Definition at line 6 of file TritonToolConfig.py.

9 name="TritonTool", **kwargs):
10 """Configure TritonTool in Control/AthOnnx/AthTritonComps/src"""
11
12 acc = ComponentAccumulator()
13
14 kwargs.setdefault("ModelName", model_name)
15 kwargs.setdefault("URL", url)
16 kwargs.setdefault("Port", port)
17 kwargs.setdefault("ModelVersion", model_version)
18 kwargs.setdefault("ClientTimeout", timeout)
19 kwargs.setdefault("UseSSL", ssl) # Default to not using SSL
20
21 if port == 443: # If the port is 443, that's typically used for HTTPS.
22 kwargs.setdefault("UseSSL", True)
23
24
25 acc.setPrivateTools(CompFactory.AthInfer.TritonTool(name=name, **kwargs))
26 return acc