def __init__()

in tfx_addons/schema_curation/component/component.py [0:0]

• 13 lines of code
• 1 McCabe index (conditional complexity)

  def __init__(
      self,
      schema: types.Channel,
      module_file: Optional[Union[Text, data_types.RuntimeParameter]] = None,
      module_path: Optional[Union[Text, data_types.RuntimeParameter]] = None,
      schema_fn: Optional[Union[Text, data_types.RuntimeParameter]] = None):
    """Construct a SchemaCurationComponent.

        Args:
          schema: A dictionary that containes the schema generated by
            SchemaGen component of tfx
          custom_schema: A dictionary that contains the schema after curation
            by the custom schema curation component
        """

    custom_schema = types.Channel(type=standard_artifacts.Schema)

    spec = SchemaCurationSpec(schema=schema,
                              custom_schema=custom_schema,
                              module_file=module_file,
                              module_path=module_path,
                              schema_fn=schema_fn)
    super().__init__(spec=spec)