ml_switcheroo.plugins.topk

Plugin for TopK Output Adaptation.

Functions

transform_topk(→ libcst.CSTNode)

Module Contents

ml_switcheroo.plugins.topk.transform_topk(node: libcst.Call, ctx: ml_switcheroo.core.hooks.HookContext) → libcst.CSTNode