rollout_to_netcdf
=================

.. py:module:: rollout_to_netcdf


Attributes
----------

.. autoapisummary::

   rollout_to_netcdf.logger
   rollout_to_netcdf.description


Classes
-------

.. autoapisummary::

   rollout_to_netcdf.ForecastProcessor


Functions
---------

.. autoapisummary::

   rollout_to_netcdf.predict


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

.. py:data:: logger

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

   .. py:attribute:: conf


   .. py:attribute:: device


   .. py:attribute:: batch_size


   .. py:attribute:: ensemble_size


   .. py:attribute:: lead_time_periods


   .. 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:data:: description
   :value: 'Rollout AI-NWP forecasts'