ml_switcheroo.plugins.topk
==========================

.. py:module:: ml_switcheroo.plugins.topk

.. autoapi-nested-parse::

   Plugin for TopK Output Adaptation.



Functions
---------

.. autoapisummary::

   ml_switcheroo.plugins.topk.transform_topk


Module Contents
---------------

.. py:function:: transform_topk(node: libcst.Call, ctx: ml_switcheroo.core.hooks.HookContext) -> libcst.CSTNode