aggregator_agent: fix doc build by aliasing G3 classes.

simonsobs · Nov 6, 2019 · 13f329b · 13f329b
1 parent 9aa8917
commit 13f329b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/agents/aggregator/aggregator_agent.py b/agents/aggregator/aggregator_agent.py
@@ -7,7 +7,10 @@
 if os.getenv('OCS_DOC_BUILD') != 'True':
     from spt3g import core
     import so3g
-
+    G3Module = core.G3Module
+else:
+    # Alias classes that are needed for clean import in docs build.
+    G3Module = object
 
 def generate_id(hksess):
     """
@@ -199,7 +202,7 @@ def to_frame(self, hksess=None, clear=False):
         return frame
 
 
-class G3FileRotator(core.G3Module):
+class G3FileRotator(G3Module):
     """
     G3 module which handles file rotation.
     After time_per_file has elapsed, the rotator will end that file and create

diff --git a/docs/aggregator.rst b/docs/aggregator.rst
@@ -168,5 +168,5 @@ should exist and belong to the `ocs` user prior to contianer startup.
 API
 ---
 
-.. autoclass:: agents.aggregator.aggregator_agent.DataAggregator
+.. autoclass:: agents.aggregator.aggregator_agent.AggregatorAgent
     :members: initialize, start_aggregate, add_feed