Add utils documentation

pockerman · pockerman · commit 47466de873f0 · 2022-06-03T20:22:18.000+01:00
diff --git a/docs/source/API/utils/episode_info.rst b/docs/source/API/utils/episode_info.rst
@@ -0,0 +1,6 @@
+episode\_info
+=============
+
+.. automodule:: episode_info
+
+.. autoclass:: EpisodeInfo
diff --git a/docs/source/API/utils/function_wraps.rst b/docs/source/API/utils/function_wraps.rst
@@ -0,0 +1,5 @@
+function\_wraps
+===============
+
+.. automodule:: function_wraps
+   :members: time_func, time_func_wrapper
diff --git a/docs/source/API/utils/iteration_control.rst b/docs/source/API/utils/iteration_control.rst
@@ -0,0 +1,7 @@
+iteration\_control
+==================
+
+.. automodule:: iteration_control
+
+.. autoclass:: IterationControl
+   :members: __init__, continue_itrs, 
diff --git a/docs/source/API/utils/replay_buffer.rst b/docs/source/API/utils/replay_buffer.rst
@@ -0,0 +1,7 @@
+replay\_buffer
+==============
+
+.. automodule:: replay_buffer
+
+.. autoclass:: ReplayBuffer
+   :members: __init__, __len__, __getitem__, add, sample, get_item_as_torch_tensor, reinitialize
diff --git a/src/utils/experience_buffer.py b/src/utils/experience_buffer.py
