Fixup output collector related code.

vishalbelsare · Sep 24, 2021 · 8608855 · 8608855
1 parent 8ce84b7
commit 8608855
Show file tree

Hide file tree

Showing 7 changed files with 5 additions and 6 deletions.
diff --git a/gQuant/plugins/gquant_plugin/notebooks/02_single_stock_trade.ipynb b/gQuant/plugins/gquant_plugin/notebooks/02_single_stock_trade.ipynb
@@ -86,7 +86,6 @@
       "    in4: sharpe_ratio.sharpe_out\n",
       "    in5: cumulative_return.cum_return\n",
       "    in6: stock_data.cudf_out\n",
-      "  module: rapids_modules\n",
       "  type: Output_Collector\n"
      ]
     }

diff --git a/gQuant/plugins/gquant_plugin/taskgraphs/get_return_feature.gq.yaml b/gQuant/plugins/gquant_plugin/taskgraphs/get_return_feature.gq.yaml
@@ -9,7 +9,6 @@
   inputs:
     in1: stock_data.cudf_out
     in2: add_return_feature.stock_out
-  module: rapids_modules
   type: Output_Collector
 - conf:
   - column: volume

diff --git a/gQuant/plugins/gquant_plugin/taskgraphs/portfolio_trade.gq.yaml b/gQuant/plugins/gquant_plugin/taskgraphs/portfolio_trade.gq.yaml
@@ -74,5 +74,4 @@
   inputs:
     in1: cumulative_return.cum_return
     in2: sharpe_ratio.sharpe_out
-  module: rapids_modules
   type: Output_Collector
diff --git a/gQuant/plugins/gquant_plugin/taskgraphs/simple_trade.gq.yaml b/gQuant/plugins/gquant_plugin/taskgraphs/simple_trade.gq.yaml
@@ -27,7 +27,6 @@
     in4: sharpe_ratio.sharpe_out
     in5: cumulative_return.cum_return
     in6: stock_data.cudf_out
-  module: rapids_modules
   type: Output_Collector
 - conf:
     keys:

diff --git a/gQuant/plugins/gquant_plugin/taskgraphs/tutorial_intro.gq.yaml b/gQuant/plugins/gquant_plugin/taskgraphs/tutorial_intro.gq.yaml
@@ -9,7 +9,6 @@
   inputs:
     in1: csv_output2.df_out
     in2: csv_output1.df_out
-  module: rapids_modules
   type: Output_Collector
 - conf:
     input:

diff --git a/gQuant/plugins/gquant_plugin/tests/unit/test_performance.py b/gQuant/plugins/gquant_plugin/tests/unit/test_performance.py
@@ -49,7 +49,8 @@ def setUp(self):
             'ports_setup.taskGraph.py': 5,
             'ports_setup._node_flow.py': 320,
             'ports_setup.template_node_mixin.py': 77,
-            'ports_setup_ext._node_taskgraph_extension_mixin.py': 77
+            'ports_setup_ext._node_taskgraph_extension_mixin.py': 77,
+            'ports_setup.output_collector_node.py': 5
         }
 
         self.meta_setup_ref = {
@@ -63,6 +64,7 @@ def setUp(self):
             'meta_setup._node_flow.py': 177,
             'meta_setup.template_node_mixin.py': 47,
             'meta_setup_ext._node_taskgraph_extension_mixin.py': 47,
+            'meta_setup.output_collector_node.py': 5
         }
 
         tgraphpath = str(topdir) + \

diff --git a/greenflow/greenflow/dataframe_flow/output_collector_node.py b/greenflow/greenflow/dataframe_flow/output_collector_node.py
@@ -4,6 +4,7 @@
 
 __all__ = ['Output_Collector', 'OUTPUT_ID', 'OUTPUT_TYPE']
 
+
 class Output_Collector(Node):
     def meta_setup(self):
         return super().meta_setup()
@@ -17,6 +18,7 @@ def conf_schema(self):
     def process(self, inputs):
         return super().process(inputs)
 
+
 # TODO: DO NOT RELY ON special OUTPUT_ID.
 # OUTPUT_ID = 'f291b900-bd19-11e9-aca3-a81e84f29b0f_uni_output'
 OUTPUT_ID = 'collector_id_fd9567b6'