AthExTriton_test Namespace Reference

Functions
	AthExTritonCfg (flags, name="AthExTritonExample", **kwargs)

Variables
	flags = initConfigFlags()
	acc = MainServicesCfg(flags)
	withDetails
	True
	summariseProps

Function Documentation

AthExTritonCfg()

AthExTriton_test.AthExTritonCfg	(		flags,
			name = "AthExTritonExample",
		**	kwargs )

Definition at line 8 of file AthExTriton_test.py.

def AthExTritonCfg(flags, name="AthExTritonExample", **kwargs):
    acc = ComponentAccumulator()
    from AthTritonComps.TritonToolConfig import TritonToolCfg
    try:
        kwargs.setdefault("InferenceTool", acc.popToolsAndMerge(
            TritonToolCfg(flags, "MNIST_testModel", "localhost",
                          name="EvaluateModelTritonTool")
        ))
    except RuntimeError as e:
        import sys
        from AthenaCommon.Logging import log as msg
        msg.warning(e)
        sys.exit(2)  # indicate test is skipped, not failed
 
    input_data = "dev/MLTest/2020-03-31/t10k-images-idx3-ubyte"
    kwargs.setdefault("BatchSize", 2)
    kwargs.setdefault("InputDataPixel", input_data)
    kwargs.setdefault("OutputLevel", Constants.DEBUG)
    acc.addEventAlgo(
        CompFactory.AthInfer.ExampleMLInferenceWithTriton(name, **kwargs))
 
    return acc
 
 

Variable Documentation

acc

AthExTriton_test.acc = MainServicesCfg(flags)

Definition at line 42 of file AthExTriton_test.py.

flags

AthExTriton_test.flags = initConfigFlags()

Definition at line 39 of file AthExTriton_test.py.

summariseProps

AthExTriton_test.summariseProps

Definition at line 44 of file AthExTriton_test.py.

True

AthExTriton_test.True

Definition at line 44 of file AthExTriton_test.py.

withDetails

AthExTriton_test.withDetails

Definition at line 44 of file AthExTriton_test.py.