module file parameters added to component spec class

component/component.py

@@ -32,9 +32,9 @@ class SchemaCurationSpec(types.ComponentSpec):
   """ComponentSpec for TFX Schema Curation Custom Component."""
 
   PARAMETERS = {
-      # These are parameters that will be passed in the call to
-      # create an instance of this component.
-      # 'name': ExecutionParameter(type=Text),
+      'module_file': ExecutionParameter(type=str, optional=True),
+      'module_path': ExecutionParameter(type=str, optional=True),
+      'preprocessing_fn': ExecutionParameter(type=str, optional=True),
       'exclude_splits': ExecutionParameter(type=str, optional=True),
   }
   INPUTS = {
@@ -49,6 +49,12 @@ class SchemaCurationSpec(types.ComponentSpec):
 class SchemaCuration(base_component.BaseComponent):
   """Custom TFX Schema Curation Component.
 
+  The SchemaCuration component is used to apply user code to a chema generated by SchemaGen
+  in order to curate the schema based on domain knowledge. 
+
+  Component `outputs` contains:
+   - `output_schema`: Channel of type `standard_artifact.Schema`
+
   Current progress : 
     - Accepts schema, outputs the same schema
   """