credit.preblock.sqrt
====================

.. py:module:: credit.preblock.sqrt


Classes
-------

.. autoapisummary::

   credit.preblock.sqrt.SqrtTransform


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

.. py:class:: SqrtTransform(variables: list[str], data_types: list[str] = None)

   Bases: :py:obj:`credit.preblock.base.BasePreblock`


   Applies a sqrt transformation to specified variables in a batch dict.

   Expected dict structure:
       batch[source][data_type]['source/var_type/var_shape/var_name']

   Config example:
       type: "sqrt_transform"
       args:
           variables:
               - 'ERA5/prognostic/3D/Q'
           data_types:     # optional, defaults to ['input', 'target']
               - 'input'
               - 'target'


   .. py:attribute:: variables


   .. py:attribute:: data_types
      :value: ['input', 'target']



   .. py:method:: forward(batch: dict) -> dict