Merge pull request #122 from NVIDIA/develop

Fixed a bug in the split node. The col_from_inport was only defined inside the if condition. When adding a new node, the col_from_inport was not defined and produced "not defined' error.

Author: avolkov1

gQuant/plugins/gquant_plugin/greenflow_gquant_plugin/ml/splitDataNode.py

@@ -56,6 +56,7 @@ def ports_setup(self):
 
     def meta_setup(self):
         cols_required = {}
+        col_from_inport = {}
         required = {
             self.INPUT_PORT_NAME: cols_required
         }
@@ -64,7 +65,6 @@ def meta_setup(self):
             col_inport = input_meta[self.INPUT_PORT_NAME]
             if 'target' in self.conf:
                 target_col = self.conf['target']
-                col_from_inport = {}
                 for i in sorted(col_inport.keys()):
                     if i != target_col:
                         col_from_inport[i] = col_inport[i]

gQuant/plugins/gquant_plugin/setup.py

@@ -2,6 +2,7 @@
 
 setup(
     name='greenflow_gquant_plugin',
+    version='0.0.1',
     install_requires=[
         "bqplot", "tables", "ray[tune]"
     ],