rollout_samudra
===============

.. py:module:: rollout_samudra


Attributes
----------

.. autoapisummary::

   rollout_samudra.logger


Classes
-------

.. autoapisummary::

   rollout_samudra.ForecastProcessor


Functions
---------

.. autoapisummary::

   rollout_samudra.predict
   rollout_samudra.main


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

.. py:data:: logger

.. py:class:: ForecastProcessor(conf, state_transformer, device)

   .. py:attribute:: conf


   .. py:attribute:: device


   .. py:attribute:: batch_size


   .. py:attribute:: ensemble_size


   .. py:attribute:: lead_time_periods


   .. py:attribute:: state_transformer


   .. py:attribute:: latlons


   .. py:attribute:: meta_data


   .. py:method:: process(y_pred, forecast_step, forecast_count, datetimes, save_datetimes)


.. py:function:: predict(rank, world_size, conf, p)

.. py:function:: main()