# 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
from synapse.ml.services.bing._BingImageSearch import _BingImageSearch
from synapse.ml.stages import Lambda
from pyspark.ml.common import inherit_doc
from pyspark.sql import SparkSession
[docs]@inherit_doc
class BingImageSearch(_BingImageSearch):
[docs] def setQuery(self, value):
self._java_obj = self._java_obj.setQuery(value)
return self
[docs] def setQueryCol(self, value):
self._java_obj = self._java_obj.setQueryCol(value)
return self
[docs] def setMarket(self, value):
self._java_obj = self._java_obj.setMarket(value)
return self
[docs] def setMarketCol(self, value):
self._java_obj = self._java_obj.setMarketCol(value)
return self
[docs] @staticmethod
def downloadFromUrls(pathCol, bytesCol, concurrency, timeout):
bis = (
SparkSession.builder.getOrCreate()._jvm.com.microsoft.azure.synapse.ml.services.bing.BingImageSearch
)
return Lambda._from_java(
bis.downloadFromUrls(pathCol, bytesCol, concurrency, timeout),
)