python.TritonToolConfig Namespace Reference

Functions
def	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()

def 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.

def TritonToolCfg(flags, model_name: str, url: str,
                  port: int = 8001, model_version: str = "",
                  timeout: float = 0., ssl: bool = False,
                  name="TritonTool", **kwargs):
    """Configure TritonTool in Control/AthOnnx/AthTritonComps/src"""
 
    acc = ComponentAccumulator()
 
    kwargs.setdefault("ModelName", model_name)
    kwargs.setdefault("URL", url)
    kwargs.setdefault("Port", port)
    kwargs.setdefault("ModelVersion", model_version)
    kwargs.setdefault("ClientTimeout", timeout)
    kwargs.setdefault("UseSSL", ssl)  # Default to not using SSL
 
    if port == 443: # If the port is 443, that's typically used for HTTPS.
        kwargs.setdefault("UseSSL", True)
 
 
    acc.setPrivateTools(CompFactory.AthInfer.TritonTool(name=name, **kwargs))
    return acc