TRex package contains functions that do not work as expected.

Description

The trex.df_preprocessing module contains two functions that do not perform as expected: clean_df and clean_for_display.

The clean_df function includes an unintended list comprehension that fails to properly clean up the dataframe by removing columns.
This issue also affects the clean_for_display function.

Environment

TensorRT Version: 8.2

NVIDIA GPU: Jetson Nano developer kit (4 GB RAM), Jetpack 4.6.4

NVIDIA Driver Version: Jetson Nano developer kit (4 GB RAM), Jetpack 4.6.4

CUDA Version: 10.2.300

CUDNN Version: 8.2.1.32

Operating System: Jetpack 4.6.4, L4T 32.7.4

Python Version (if applicable): 3.10.15

Baremetal or Container (if so, version): Baremetal

Steps To Reproduce

• Generate an Engine Plan as described in tools/experimental/trt-engine-explorer/notebooks/tutorial.ipynb.
• Check which columns should be deleted with clean_for_display function.

As an alternative execute the following block of code, which includes the list comprehension copied from clean_df function.

from trex.df_preprocessing import layer_attributes
columns = set([col for col_list in layer_attributes.keys() for col in col_list])
print(columns)

Result is:
     {'2', 'C', 'S', 'R', 'P', 'W', 'V', 'c', 'a', 'g', 'f', 'e', 'i', 'h', 'o', 'n', 'm', 'l', 's', 'r', 'v', 'u', 't', 'z'}

Should be:
    ['subtype', 'Groups', 'OutMaps', 'Stride', 'Kernel', 'HasBias', 'Activation', 'HasReLU', 'PaddingMode', 'PrePadding', 'PostPadding', 'Dilation', 'AllowSparse', 'Weights', 'Bias', 'Dimensions', 'ConvolutionTacticIndex', 'Operations', 'NbOperations', 'Operations', 'NbOperations', 'NbInputArgs', 'InputArgs', 'NbOutputVars', 'OutputVars', 'NbLiterals', 'Literals', 'NbParams', 'Params', 'Instructions', 'NbInstructions', 'NbInputs', 'InputArgs', 'NbOutputVars', 'OutputVars', 'NbLiterals', 'Literals', 'NbParams', 'Params', 'Origin', 'PoolingType', 'WindowSize', 'AverageCountExcludesPadding', 'PaddingMode', 'Stride', 'PrePadding', 'PostPadding', 'BlendFactor', 'Mode', 'Shift', 'Scale', 'Power', 'Activation', 'ChannelAxis', 'FirstTranspose', 'SecondTranspose', 'Reshape', 'ZeroIsPlaceholder', 'ParameterSubType', 'ResizeMode', 'ResizeScales', 'NNRounding', 'Start', 'CoordTransform', 'ResizeSelector', 'Start', 'Stride', 'Size', 'Mode', 'negativeInfinityPadding', 'ParameterType', 'weights', 'dimensions', 'TacticValue']

Start in the trex.df_preprocessing module by examining clean_df and clean_for_display, then reproduce the issue with the list comprehension shown in the report or tools/experimental/trt-engine-explorer/notebooks/tutorial.ipynb. Done means both functions identify and remove the expected dataframe columns rather than individual characters.