Source code for synapse.ml.cognitive.AzureSearchWriter

# Copyright (C) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See LICENSE in project root for information.

import sys

if sys.version >= '3':
    basestring = str

import pyspark
from pyspark import SparkContext
from pyspark import sql
from pyspark.ml.param.shared import *
from pyspark.sql import DataFrame

[docs]def streamToAzureSearch(df, **options): jvm = SparkContext.getOrCreate()._jvm writer = jvm.com.microsoft.azure.synapse.ml.cognitive.AzureSearchWriter return writer.stream(df._jdf, options)
setattr(pyspark.sql.DataFrame, 'streamToAzureSearch', streamToAzureSearch)
[docs]def writeToAzureSearch(df, **options): jvm = SparkContext.getOrCreate()._jvm writer = jvm.com.microsoft.azure.synapse.ml.cognitive.AzureSearchWriter writer.write(df._jdf, options)
setattr(pyspark.sql.DataFrame, 'writeToAzureSearch', writeToAzureSearch)