scaler
======

.. py:module:: scaler


Attributes
----------

.. autoapisummary::

   scaler.scalers


Functions
---------

.. autoapisummary::

   scaler.main
   scaler.fit_era5_scaler_times
   scaler.transform_era5_times
   scaler.fit_scaled_era5_time_residuals


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

.. py:data:: scalers

.. py:function:: main()

.. py:function:: fit_era5_scaler_times(times, rank, era5_file_dir=None, vars_3d=None, vars_surf=None, scaler_type='quantile', scaler_config=None)

.. py:function:: transform_era5_times(times, rank, scaler_file=None, era5_file_dir=None, vars_3d=None, vars_surf=None, scaler_type=None, out_dir='/glade/derecho/scratch/dgagne/era5_quantile/', var_encoding=None)

.. py:function:: fit_scaled_era5_time_residuals(times, rank, era5_file_dir=None, dt=1, scaler_type='standard', scaler_config=None)

   Fit scalers to distributions of time differences for each variable.

   :param times: List or Series of times
   :param rank: MPI rank
   :type rank: int
   :param era5_file_dir: Path to era5 scaled files
   :type era5_file_dir: str
   :param dt: number of hours difference
   :type dt: int
   :param scaler_type: standard or quantile
   :type scaler_type: str
   :param scaler_config: kwargs for the scaler obj
   :type scaler_config: dict

   :returns: 3D scaler, surface scaler