def writer()

in pyspark_huggingface/huggingface_sink.py [0:0]

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

    def writer(self, schema: StructType, overwrite: bool) -> "HuggingFaceDatasetsWriter":
        return HuggingFaceDatasetsWriter(
            repo_id=self.repo_id,
            path_in_repo=self.path_in_repo,
            split=self.split,
            revision=self.revision,
            schema=schema,
            overwrite=overwrite,
            token=self.token,
            endpoint=self.endpoint,
            **self.kwargs,
        )