def write_stats_and_schema()

in spotify_tensorflow/tfx/tfdv.py [0:0]

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

    def write_stats_and_schema(self,
                               pipeline_args,
                               infer_feature_shape=False
                               ):
        # type: (List[str], bool) -> None
        stats = self.write_stats(pipeline_args)
        if not self.schema:
            logger.warning(
                "Inferring a new schema for this dataset. If you want to use an existing schema, "
                "provide a value for schema_path in the constructor."
            )
            new_schema = tfdv.infer_schema(stats, infer_feature_shape=infer_feature_shape)
            self.schema = new_schema
        self.upload_schema()