Skip to content
Snippets Groups Projects
  1. Mar 18, 2017
    • Sean Owen's avatar
      [SPARK-16599][CORE] java.util.NoSuchElementException: None.get at at... · 54e61df2
      Sean Owen authored
      [SPARK-16599][CORE] java.util.NoSuchElementException: None.get at at org.apache.spark.storage.BlockInfoManager.releaseAllLocksForTask
      
      ## What changes were proposed in this pull request?
      
      Avoid None.get exception in (rare?) case that no readLocks exist
      Note that while this would resolve the immediate cause of the exception, it's not clear it is the root problem.
      
      ## How was this patch tested?
      
      Existing tests
      
      Author: Sean Owen <sowen@cloudera.com>
      
      Closes #17290 from srowen/SPARK-16599.
      54e61df2
    • Takeshi Yamamuro's avatar
      [SPARK-19896][SQL] Throw an exception if case classes have circular references in toDS · ccba622e
      Takeshi Yamamuro authored
      ## What changes were proposed in this pull request?
      If case classes have circular references below, it throws StackOverflowError;
      ```
      scala> :pasge
      case class classA(i: Int, cls: classB)
      case class classB(cls: classA)
      
      scala> Seq(classA(0, null)).toDS()
      java.lang.StackOverflowError
        at scala.reflect.internal.Symbols$Symbol.info(Symbols.scala:1494)
        at scala.reflect.runtime.JavaMirrors$JavaMirror$$anon$1.scala$reflect$runtime$SynchronizedSymbols$SynchronizedSymbol$$super$info(JavaMirrors.scala:66)
        at scala.reflect.runtime.SynchronizedSymbols$SynchronizedSymbol$$anonfun$info$1.apply(SynchronizedSymbols.scala:127)
        at scala.reflect.runtime.SynchronizedSymbols$SynchronizedSymbol$$anonfun$info$1.apply(SynchronizedSymbols.scala:127)
        at scala.reflect.runtime.Gil$class.gilSynchronized(Gil.scala:19)
        at scala.reflect.runtime.JavaUniverse.gilSynchronized(JavaUniverse.scala:16)
        at scala.reflect.runtime.SynchronizedSymbols$SynchronizedSymbol$class.gilSynchronizedIfNotThreadsafe(SynchronizedSymbols.scala:123)
        at scala.reflect.runtime.JavaMirrors$JavaMirror$$anon$1.gilSynchronizedIfNotThreadsafe(JavaMirrors.scala:66)
        at scala.reflect.runtime.SynchronizedSymbols$SynchronizedSymbol$class.info(SynchronizedSymbols.scala:127)
        at scala.reflect.runtime.JavaMirrors$JavaMirror$$anon$1.info(JavaMirrors.scala:66)
        at scala.reflect.internal.Mirrors$RootsBase.getModuleOrClass(Mirrors.scala:48)
        at scala.reflect.internal.Mirrors$RootsBase.getModuleOrClass(Mirrors.scala:45)
        at scala.reflect.internal.Mirrors$RootsBase.getModuleOrClass(Mirrors.scala:45)
        at scala.reflect.internal.Mirrors$RootsBase.getModuleOrClass(Mirrors.scala:45)
        at scala.reflect.internal.Mirrors$RootsBase.getModuleOrClass(Mirrors.scala:45)
      ```
      This pr added code to throw UnsupportedOperationException in that case as follows;
      ```
      scala> :paste
      case class A(cls: B)
      case class B(cls: A)
      
      scala> Seq(A(null)).toDS()
      java.lang.UnsupportedOperationException: cannot have circular references in class, but got the circular reference of class B
        at org.apache.spark.sql.catalyst.ScalaReflection$.org$apache$spark$sql$catalyst$ScalaReflection$$serializerFor(ScalaReflection.scala:627)
        at org.apache.spark.sql.catalyst.ScalaReflection$$anonfun$9.apply(ScalaReflection.scala:644)
        at org.apache.spark.sql.catalyst.ScalaReflection$$anonfun$9.apply(ScalaReflection.scala:632)
        at scala.collection.TraversableLike$$anonfun$flatMap$1.apply(TraversableLike.scala:241)
        at scala.collection.TraversableLike$$anonfun$flatMap$1.apply(TraversableLike.scala:241)
        at scala.collection.immutable.List.foreach(List.scala:381)
        at scala.collection.TraversableLike$class.flatMap(TraversableLike.scala:241)
      ```
      
      ## How was this patch tested?
      Added tests in `DatasetSuite`.
      
      Author: Takeshi Yamamuro <yamamuro@apache.org>
      
      Closes #17318 from maropu/SPARK-19896.
      ccba622e
    • wangzhenhua's avatar
      [SPARK-19915][SQL] Exclude cartesian product candidates to reduce the search space · c083b6b7
      wangzhenhua authored
      ## What changes were proposed in this pull request?
      
      We have some concerns about removing size in the cost model [in the previous pr](https://github.com/apache/spark/pull/17240). It's a tradeoff between code structure and algorithm completeness. I tend to keep the size and thus create this new pr without changing cost model.
      
      What this pr does:
      1. We only consider consecutive inner joinable items, thus excluding cartesian products in reordering procedure. This significantly reduces the search space and memory overhead of memo. Otherwise every combination of items will exist in the memo.
      2. This pr also includes a bug fix: if a leaf item is a project(_, child), current solution will miss the project.
      
      ## How was this patch tested?
      
      Added test cases.
      
      Author: wangzhenhua <wangzhenhua@huawei.com>
      
      Closes #17286 from wzhfy/joinReorder3.
      c083b6b7
  2. Mar 17, 2017
    • Jacek Laskowski's avatar
      [SQL][MINOR] Fix scaladoc for UDFRegistration · 6326d406
      Jacek Laskowski authored
      ## What changes were proposed in this pull request?
      
      Fix scaladoc for UDFRegistration
      
      ## How was this patch tested?
      
      local build
      
      Author: Jacek Laskowski <jacek@japila.pl>
      
      Closes #17337 from jaceklaskowski/udfregistration-scaladoc.
      6326d406
    • Kunal Khamar's avatar
      [SPARK-19873][SS] Record num shuffle partitions in offset log and enforce in next batch. · 3783539d
      Kunal Khamar authored
      ## What changes were proposed in this pull request?
      
      If the user changes the shuffle partition number between batches, Streaming aggregation will fail.
      
      Here are some possible cases:
      
      - Change "spark.sql.shuffle.partitions"
      - Use "repartition" and change the partition number in codes
      - RangePartitioner doesn't generate deterministic partitions. Right now it's safe as we disallow sort before aggregation. Not sure if we will add some operators using RangePartitioner in future.
      
      ## How was this patch tested?
      
      - Unit tests
      - Manual tests
        - forward compatibility tested by using the new `OffsetSeqMetadata` json with Spark v2.1.0
      
      Author: Kunal Khamar <kkhamar@outlook.com>
      
      Closes #17216 from kunalkhamar/num-partitions.
      3783539d
    • Takeshi Yamamuro's avatar
      [SPARK-19967][SQL] Add from_json in FunctionRegistry · 7de66bae
      Takeshi Yamamuro authored
      ## What changes were proposed in this pull request?
      This pr added entries in `FunctionRegistry` and supported `from_json` in SQL.
      
      ## How was this patch tested?
      Added tests in `JsonFunctionsSuite` and `SQLQueryTestSuite`.
      
      Author: Takeshi Yamamuro <yamamuro@apache.org>
      
      Closes #17320 from maropu/SPARK-19967.
      7de66bae
    • Andrew Ray's avatar
      [SPARK-18847][GRAPHX] PageRank gives incorrect results for graphs with sinks · bfdeea5c
      Andrew Ray authored
      ## What changes were proposed in this pull request?
      
      Graphs with sinks (vertices with no outgoing edges) don't have the expected rank sum of n (or 1 for personalized). We fix this by normalizing to the expected sum at the end of each implementation.
      
      Additionally this fixes the dynamic version of personal pagerank which gave incorrect answers that were not detected by existing unit tests.
      
      ## How was this patch tested?
      
      Revamped existing and additional unit tests with reference values (and reproduction code) from igraph and NetworkX.
      
      Note that for comparison on personal pagerank we use the arpack algorithm in igraph as prpack (the  current default) redistributes rank to all vertices uniformly instead of just to the personalization source. We could take the alternate convention (redistribute rank to all vertices uniformly) but that would involve more extensive changes to the algorithms (the dynamic version would no longer be able to use Pregel).
      
      Author: Andrew Ray <ray.andrew@gmail.com>
      
      Closes #16483 from aray/pagerank-sink2.
      bfdeea5c
    • Shixiong Zhu's avatar
      [SPARK-19986][TESTS] Make pyspark.streaming.tests.CheckpointTests more stable · 376d7821
      Shixiong Zhu authored
      ## What changes were proposed in this pull request?
      
      Sometimes, CheckpointTests will hang on a busy machine because the streaming jobs are too slow and cannot catch up. I observed the scheduled delay was keeping increasing for dozens of seconds locally.
      
      This PR increases the batch interval from 0.5 seconds to 2 seconds to generate less Spark jobs. It should make `pyspark.streaming.tests.CheckpointTests` more stable. I also replaced `sleep` with `awaitTerminationOrTimeout` so that if the streaming job fails, it will also fail the test.
      
      ## How was this patch tested?
      
      Jenkins
      
      Author: Shixiong Zhu <shixiong@databricks.com>
      
      Closes #17323 from zsxwing/SPARK-19986.
      376d7821
    • Sital Kedia's avatar
      [SPARK-13369] Add config for number of consecutive fetch failures · 7b5d873a
      Sital Kedia authored
      The previously hardcoded max 4 retries per stage is not suitable for all cluster configurations. Since spark retries a stage at the sign of the first fetch failure, you can easily end up with many stage retries to discover all the failures. In particular, two scenarios this value should change are (1) if there are more than 4 executors per node; in that case, it may take 4 retries to discover the problem with each executor on the node and (2) during cluster maintenance on large clusters, where multiple machines are serviced at once, but you also cannot afford total cluster downtime. By making this value configurable, cluster managers can tune this value to something more appropriate to their cluster configuration.
      
      Unit tests
      
      Author: Sital Kedia <skedia@fb.com>
      
      Closes #17307 from sitalkedia/SPARK-13369.
      7b5d873a
    • Andrew Ray's avatar
      [SPARK-19882][SQL] Pivot with null as a distinct pivot value throws NPE · 13538cf3
      Andrew Ray authored
      ## What changes were proposed in this pull request?
      
      Allows null values of the pivot column to be included in the pivot values list without throwing NPE
      
      Note this PR was made as an alternative to #17224 but preserves the two phase aggregate operation that is needed for good performance.
      
      ## How was this patch tested?
      
      Additional unit test
      
      Author: Andrew Ray <ray.andrew@gmail.com>
      
      Closes #17226 from aray/pivot-null.
      13538cf3
  3. Mar 16, 2017
    • Reynold Xin's avatar
      [SPARK-19987][SQL] Pass all filters into FileIndex · 8537c00e
      Reynold Xin authored
      ## What changes were proposed in this pull request?
      This is a tiny teeny refactoring to pass data filters also to the FileIndex, so FileIndex can have a more global view on predicates.
      
      ## How was this patch tested?
      Change should be covered by existing test cases.
      
      Author: Reynold Xin <rxin@databricks.com>
      
      Closes #17322 from rxin/SPARK-19987.
      8537c00e
    • Joseph K. Bradley's avatar
      [SPARK-19635][ML] DataFrame-based API for chi square test · 4c320054
      Joseph K. Bradley authored
      ## What changes were proposed in this pull request?
      
      Wrapper taking and return a DataFrame
      
      ## How was this patch tested?
      
      Copied unit tests from RDD-based API
      
      Author: Joseph K. Bradley <joseph@databricks.com>
      
      Closes #17110 from jkbradley/df-hypotests.
      4c320054
    • Liwei Lin's avatar
      [SPARK-19721][SS] Good error message for version mismatch in log files · 2ea214dd
      Liwei Lin authored
      ## Problem
      
      There are several places where we write out version identifiers in various logs for structured streaming (usually `v1`). However, in the places where we check for this, we throw a confusing error message.
      
      ## What changes were proposed in this pull request?
      
      This patch made two major changes:
      1. added a `parseVersion(...)` method, and based on this method, fixed the following places the way they did version checking (no other place needed to do this checking):
      ```
      HDFSMetadataLog
        - CompactibleFileStreamLog  ------------> fixed with this patch
          - FileStreamSourceLog  ---------------> inherited the fix of `CompactibleFileStreamLog`
          - FileStreamSinkLog  -----------------> inherited the fix of `CompactibleFileStreamLog`
        - OffsetSeqLog  ------------------------> fixed with this patch
        - anonymous subclass in KafkaSource  ---> fixed with this patch
      ```
      
      2. changed the type of `FileStreamSinkLog.VERSION`, `FileStreamSourceLog.VERSION` etc. from `String` to `Int`, so that we can identify newer versions via `version > 1` instead of `version != "v1"`
          - note this didn't break any backwards compatibility -- we are still writing out `"v1"` and reading back `"v1"`
      
      ## Exception message with this patch
      ```
      java.lang.IllegalStateException: Failed to read log file /private/var/folders/nn/82rmvkk568sd8p3p8tb33trw0000gn/T/spark-86867b65-0069-4ef1-b0eb-d8bd258ff5b8/0. UnsupportedLogVersion: maximum supported log version is v1, but encountered v99. The log file was produced by a newer version of Spark and cannot be read by this version. Please upgrade.
      	at org.apache.spark.sql.execution.streaming.HDFSMetadataLog.get(HDFSMetadataLog.scala:202)
      	at org.apache.spark.sql.execution.streaming.OffsetSeqLogSuite$$anonfun$3$$anonfun$apply$mcV$sp$2.apply(OffsetSeqLogSuite.scala:78)
      	at org.apache.spark.sql.execution.streaming.OffsetSeqLogSuite$$anonfun$3$$anonfun$apply$mcV$sp$2.apply(OffsetSeqLogSuite.scala:75)
      	at org.apache.spark.sql.test.SQLTestUtils$class.withTempDir(SQLTestUtils.scala:133)
      	at org.apache.spark.sql.execution.streaming.OffsetSeqLogSuite.withTempDir(OffsetSeqLogSuite.scala:26)
      	at org.apache.spark.sql.execution.streaming.OffsetSeqLogSuite$$anonfun$3.apply$mcV$sp(OffsetSeqLogSuite.scala:75)
      	at org.apache.spark.sql.execution.streaming.OffsetSeqLogSuite$$anonfun$3.apply(OffsetSeqLogSuite.scala:75)
      	at org.apache.spark.sql.execution.streaming.OffsetSeqLogSuite$$anonfun$3.apply(OffsetSeqLogSuite.scala:75)
      	at org.scalatest.Transformer$$anonfun$apply$1.apply$mcV$sp(Transformer.scala:22)
      	at org.scalatest.OutcomeOf$class.outcomeOf(OutcomeOf.scala:85)
      ```
      
      ## How was this patch tested?
      
      unit tests
      
      Author: Liwei Lin <lwlin7@gmail.com>
      
      Closes #17070 from lw-lin/better-msg.
      2ea214dd
    • windpiger's avatar
      [SPARK-19945][SQL] add test suite for SessionCatalog with HiveExternalCatalog · 8e8f8983
      windpiger authored
      ## What changes were proposed in this pull request?
      
      Currently `SessionCatalogSuite` is only for `InMemoryCatalog`, there is no suite for `HiveExternalCatalog`.
      And there are some ddl function is not proper to test in `ExternalCatalogSuite`, because some logic are not full implement in `ExternalCatalog`, these ddl functions are full implement in `SessionCatalog`(e.g. merge the same logic from `ExternalCatalog` up to `SessionCatalog` ).
      It is better to test it in `SessionCatalogSuite` for this situation.
      
      So we should add a test suite for `SessionCatalog` with `HiveExternalCatalog`
      
      The main change is that in `SessionCatalogSuite` add two functions:
      `withBasicCatalog` and `withEmptyCatalog`
      And replace the code like  `val catalog = new SessionCatalog(newBasicCatalog)` with above two functions
      
      ## How was this patch tested?
      add `HiveExternalSessionCatalogSuite`
      
      Author: windpiger <songjun@outlook.com>
      
      Closes #17287 from windpiger/sessioncatalogsuit.
      8e8f8983
    • Bogdan Raducanu's avatar
      [SPARK-19946][TESTING] DebugFilesystem.assertNoOpenStreams should report the... · ee91a0de
      Bogdan Raducanu authored
      [SPARK-19946][TESTING] DebugFilesystem.assertNoOpenStreams should report the open streams to help debugging
      
      ## What changes were proposed in this pull request?
      
      DebugFilesystem.assertNoOpenStreams throws an exception with a cause exception that actually shows the code line which leaked the stream.
      
      ## How was this patch tested?
      New test in SparkContextSuite to check there is a cause exception.
      
      Author: Bogdan Raducanu <bogdan@databricks.com>
      
      Closes #17292 from bogdanrdc/SPARK-19946.
      ee91a0de
    • Yuhao Yang's avatar
      [SPARK-13568][ML] Create feature transformer to impute missing values · d647aae2
      Yuhao Yang authored
      ## What changes were proposed in this pull request?
      
      jira: https://issues.apache.org/jira/browse/SPARK-13568
      It is quite common to encounter missing values in data sets. It would be useful to implement a Transformer that can impute missing data points, similar to e.g. Imputer in scikit-learn.
      Initially, options for imputation could include mean, median and most frequent, but we could add various other approaches, where possible existing DataFrame code can be used (e.g. for approximate quantiles etc).
      
      Currently this PR supports imputation for Double and Vector (null and NaN in Vector).
      ## How was this patch tested?
      
      new unit tests and manual test
      
      Author: Yuhao Yang <hhbyyh@gmail.com>
      Author: Yuhao Yang <yuhao.yang@intel.com>
      Author: Yuhao <yuhao.yang@intel.com>
      
      Closes #11601 from hhbyyh/imputer.
      d647aae2
  4. Mar 15, 2017
    • Xiao Li's avatar
      [SPARK-19830][SQL] Add parseTableSchema API to ParserInterface · 1472cac4
      Xiao Li authored
      ### What changes were proposed in this pull request?
      
      Specifying the table schema in DDL formats is needed for different scenarios. For example,
      - [specifying the schema in SQL function `from_json` using DDL formats](https://issues.apache.org/jira/browse/SPARK-19637), which is suggested by marmbrus ,
      - [specifying the customized JDBC data types](https://github.com/apache/spark/pull/16209).
      
      These two PRs need users to use the JSON format to specify the table schema. This is not user friendly.
      
      This PR is to provide a `parseTableSchema` API in `ParserInterface`.
      
      ### How was this patch tested?
      Added a test suite `TableSchemaParserSuite`
      
      Author: Xiao Li <gatorsmile@gmail.com>
      
      Closes #17171 from gatorsmile/parseDDLStmt.
      1472cac4
    • Takeshi Yamamuro's avatar
      [SPARK-19751][SQL] Throw an exception if bean class has one's own class in fields · 21f333c6
      Takeshi Yamamuro authored
      ## What changes were proposed in this pull request?
      The current master throws `StackOverflowError` in `createDataFrame`/`createDataset` if bean has one's own class in fields;
      ```
      public class SelfClassInFieldBean implements Serializable {
        private SelfClassInFieldBean child;
        ...
      }
      ```
      This pr added code to throw `UnsupportedOperationException` in that case as soon as possible.
      
      ## How was this patch tested?
      Added tests in `JavaDataFrameSuite` and `JavaDatasetSuite`.
      
      Author: Takeshi Yamamuro <yamamuro@apache.org>
      
      Closes #17188 from maropu/SPARK-19751.
      21f333c6
    • windpiger's avatar
      [SPARK-19961][SQL][MINOR] unify a erro msg when drop databse for... · fc931467
      windpiger authored
      [SPARK-19961][SQL][MINOR] unify a erro msg when drop databse for HiveExternalCatalog and InMemoryCatalog
      
      ## What changes were proposed in this pull request?
      
      unify a exception erro msg for dropdatabase when the database still have some tables for HiveExternalCatalog and InMemoryCatalog
      ## How was this patch tested?
      N/A
      
      Author: windpiger <songjun@outlook.com>
      
      Closes #17305 from windpiger/unifyErromsg.
      fc931467
    • Juliusz Sompolski's avatar
      [SPARK-19948] Document that saveAsTable uses catalog as source of truth for table existence. · 339b237d
      Juliusz Sompolski authored
      It is quirky behaviour that saveAsTable to e.g. a JDBC source with SaveMode other
      than Overwrite will nevertheless overwrite the table in the external source,
      if that table was not a catalog table.
      
      Author: Juliusz Sompolski <julek@databricks.com>
      
      Closes #17289 from juliuszsompolski/saveAsTableDoc.
      339b237d
    • Liang-Chi Hsieh's avatar
      [SPARK-19931][SQL] InMemoryTableScanExec should rewrite output partitioning... · 7d734a65
      Liang-Chi Hsieh authored
      [SPARK-19931][SQL] InMemoryTableScanExec should rewrite output partitioning and ordering when aliasing output attributes
      
      ## What changes were proposed in this pull request?
      
      Now `InMemoryTableScanExec` simply takes the `outputPartitioning` and `outputOrdering` from the associated `InMemoryRelation`'s `child.outputPartitioning` and `outputOrdering`.
      
      However, `InMemoryTableScanExec` can alias the output attributes. In this case, its `outputPartitioning` and `outputOrdering` are not correct and its parent operators can't correctly determine its data distribution.
      
      ## How was this patch tested?
      
      Jenkins tests.
      
      Please review http://spark.apache.org/contributing.html before opening a pull request.
      
      Author: Liang-Chi Hsieh <viirya@gmail.com>
      
      Closes #17175 from viirya/ensure-no-unnecessary-shuffle.
      7d734a65
    • erenavsarogullari's avatar
      [SPARK-18066][CORE][TESTS] Add Pool usage policies test coverage for FIFO & FAIR Schedulers · 046b8d4a
      erenavsarogullari authored
      ## What changes were proposed in this pull request?
      
      The following FIFO & FAIR Schedulers Pool usage cases need to have unit test coverage :
      - FIFO Scheduler just uses **root pool** so even if `spark.scheduler.pool` property is set, related pool is not created and `TaskSetManagers` are added to **root pool**.
      - FAIR Scheduler uses `default pool` when `spark.scheduler.pool` property is not set. This can be happened when
        - `Properties` object is **null**,
        - `Properties` object is **empty**(`new Properties()`),
        - **default pool** is set(`spark.scheduler.pool=default`).
      - FAIR Scheduler creates a **new pool** with **default values** when `spark.scheduler.pool` property points a **non-existent** pool. This can be happened when **scheduler allocation file** is not set or it does not contain related pool.
      ## How was this patch tested?
      
      New Unit tests are added.
      
      Author: erenavsarogullari <erenavsarogullari@gmail.com>
      
      Closes #15604 from erenavsarogullari/SPARK-18066.
      046b8d4a
    • Dongjoon Hyun's avatar
      [MINOR][CORE] Fix a info message of `prunePartitions` · 54a3697f
      Dongjoon Hyun authored
      ## What changes were proposed in this pull request?
      
      `PrunedInMemoryFileIndex.prunePartitions` shows `pruned NaN% partitions` for the following case.
      
      ```scala
      scala> Seq.empty[(String, String)].toDF("a", "p").write.partitionBy("p").saveAsTable("t1")
      
      scala> sc.setLogLevel("INFO")
      
      scala> spark.table("t1").filter($"p" === "1").select($"a").show
      ...
      17/03/13 00:33:04 INFO PrunedInMemoryFileIndex: Selected 0 partitions out of 0, pruned NaN% partitions.
      ```
      
      After this PR, the message looks like this.
      ```scala
      17/03/15 10:39:48 INFO PrunedInMemoryFileIndex: Selected 0 partitions out of 0, pruned 0 partitions.
      ```
      
      ## How was this patch tested?
      
      Pass the Jenkins with the existing tests.
      
      Author: Dongjoon Hyun <dongjoon@apache.org>
      
      Closes #17273 from dongjoon-hyun/SPARK-EMPTY-PARTITION.
      54a3697f
    • jiangxingbo's avatar
      [SPARK-19960][CORE] Move `SparkHadoopWriter` to `internal/io/` · 97cc5e5a
      jiangxingbo authored
      ## What changes were proposed in this pull request?
      
      This PR introduces the following changes:
      1. Move `SparkHadoopWriter` to `core/internal/io/`, so that it's in the same directory with `SparkHadoopMapReduceWriter`;
      2. Move `SparkHadoopWriterUtils` to a separated file.
      
      After this PR is merged, we may consolidate `SparkHadoopWriter` and `SparkHadoopMapReduceWriter`, and make the new commit protocol support the old `mapred` package's committer;
      
      ## How was this patch tested?
      
      Tested by existing test cases.
      
      Author: jiangxingbo <jiangxb1987@gmail.com>
      
      Closes #17304 from jiangxb1987/writer.
      97cc5e5a
    • Tejas Patil's avatar
      [SPARK-13450] Introduce ExternalAppendOnlyUnsafeRowArray. Change... · 02c274ea
      Tejas Patil authored
      [SPARK-13450] Introduce ExternalAppendOnlyUnsafeRowArray. Change CartesianProductExec, SortMergeJoin, WindowExec to use it
      
      ## What issue does this PR address ?
      
      Jira: https://issues.apache.org/jira/browse/SPARK-13450
      
      In `SortMergeJoinExec`, rows of the right relation having the same value for a join key are buffered in-memory. In case of skew, this causes OOMs (see comments in SPARK-13450 for more details). Heap dump from a failed job confirms this : https://issues.apache.org/jira/secure/attachment/12846382/heap-dump-analysis.png . While its possible to increase the heap size to workaround, Spark should be resilient to such issues as skews can happen arbitrarily.
      
      ## Change proposed in this pull request
      
      - Introduces `ExternalAppendOnlyUnsafeRowArray`
        - It holds `UnsafeRow`s in-memory upto a certain threshold.
        - After the threshold is hit, it switches to `UnsafeExternalSorter` which enables spilling of the rows to disk. It does NOT sort the data.
        - Allows iterating the array multiple times. However, any alteration to the array (using `add` or `clear`) will invalidate the existing iterator(s)
      - `WindowExec` was already using `UnsafeExternalSorter` to support spilling. Changed it to use the new array
      - Changed `SortMergeJoinExec` to use the new array implementation
        - NOTE: I have not changed FULL OUTER JOIN to use this new array implementation. Changing that will need more surgery and I will rather put up a separate PR for that once this gets in.
      - Changed `CartesianProductExec` to use the new array implementation
      
      #### Note for reviewers
      
      The diff can be divided into 3 parts. My motive behind having all the changes in a single PR was to demonstrate that the API is sane and supports 2 use cases. If reviewing as 3 separate PRs would help, I am happy to make the split.
      
      ## How was this patch tested ?
      
      #### Unit testing
      - Added unit tests `ExternalAppendOnlyUnsafeRowArray` to validate all its APIs and access patterns
      - Added unit test for `SortMergeExec`
        - with and without spill for inner join, left outer join, right outer join to confirm that the spill threshold config behaves as expected and output is as expected.
        - This PR touches the scanning logic in `SortMergeExec` for _all_ joins (except FULL OUTER JOIN). However, I expect existing test cases to cover that there is no regression in correctness.
      - Added unit test for `WindowExec` to check behavior of spilling and correctness of results.
      
      #### Stress testing
      - Confirmed that OOM is gone by running against a production job which used to OOM
      - Since I cannot share details about prod workload externally, created synthetic data to mimic the issue. Ran before and after the fix to demonstrate the issue and query success with this PR
      
      Generating the synthetic data
      
      ```
      ./bin/spark-shell --driver-memory=6G
      
      import org.apache.spark.sql._
      val hc = SparkSession.builder.master("local").getOrCreate()
      
      hc.sql("DROP TABLE IF EXISTS spark_13450_large_table").collect
      hc.sql("DROP TABLE IF EXISTS spark_13450_one_row_table").collect
      
      val df1 = (0 until 1).map(i => ("10", "100", i.toString, (i * 2).toString)).toDF("i", "j", "str1", "str2")
      df1.write.format("org.apache.spark.sql.hive.orc.OrcFileFormat").bucketBy(100, "i", "j").sortBy("i", "j").saveAsTable("spark_13450_one_row_table")
      
      val df2 = (0 until 3000000).map(i => ("10", "100", i.toString, (i * 2).toString)).toDF("i", "j", "str1", "str2")
      df2.write.format("org.apache.spark.sql.hive.orc.OrcFileFormat").bucketBy(100, "i", "j").sortBy("i", "j").saveAsTable("spark_13450_large_table")
      ```
      
      Ran this against trunk VS local build with this PR. OOM repros with trunk and with the fix this query runs fine.
      
      ```
      ./bin/spark-shell --driver-java-options="-XX:+HeapDumpOnOutOfMemoryError -XX:HeapDumpPath=/tmp/spark.driver.heapdump.hprof"
      
      import org.apache.spark.sql._
      val hc = SparkSession.builder.master("local").getOrCreate()
      hc.sql("SET spark.sql.autoBroadcastJoinThreshold=1")
      hc.sql("SET spark.sql.sortMergeJoinExec.buffer.spill.threshold=10000")
      
      hc.sql("DROP TABLE IF EXISTS spark_13450_result").collect
      hc.sql("""
        CREATE TABLE spark_13450_result
        AS
        SELECT
          a.i AS a_i, a.j AS a_j, a.str1 AS a_str1, a.str2 AS a_str2,
          b.i AS b_i, b.j AS b_j, b.str1 AS b_str1, b.str2 AS b_str2
        FROM
          spark_13450_one_row_table a
        JOIN
          spark_13450_large_table b
        ON
          a.i=b.i AND
          a.j=b.j
      """)
      ```
      
      ## Performance comparison
      
      ### Macro-benchmark
      
      I ran a SMB join query over two real world tables (2 trillion rows (40 TB) and 6 million rows (120 GB)). Note that this dataset does not have skew so no spill happened. I saw improvement in CPU time by 2-4% over version without this PR. This did not add up as I was expected some regression. I think allocating array of capacity of 128 at the start (instead of starting with default size 16) is the sole reason for the perf. gain : https://github.com/tejasapatil/spark/blob/SPARK-13450_smb_buffer_oom/sql/core/src/main/scala/org/apache/spark/sql/execution/ExternalAppendOnlyUnsafeRowArray.scala#L43 . I could remove that and rerun, but effectively the change will be deployed in this form and I wanted to see the effect of it over large workload.
      
      ### Micro-benchmark
      
      Two types of benchmarking can be found in `ExternalAppendOnlyUnsafeRowArrayBenchmark`:
      
      [A] Comparing `ExternalAppendOnlyUnsafeRowArray` against raw `ArrayBuffer` when all rows fit in-memory and there is no spill
      
      ```
      Array with 1000 rows:                    Best/Avg Time(ms)    Rate(M/s)   Per Row(ns)   Relative
      ------------------------------------------------------------------------------------------------
      ArrayBuffer                                   7821 / 7941         33.5          29.8       1.0X
      ExternalAppendOnlyUnsafeRowArray              8798 / 8819         29.8          33.6       0.9X
      
      Array with 30000 rows:                   Best/Avg Time(ms)    Rate(M/s)   Per Row(ns)   Relative
      ------------------------------------------------------------------------------------------------
      ArrayBuffer                                 19200 / 19206         25.6          39.1       1.0X
      ExternalAppendOnlyUnsafeRowArray            19558 / 19562         25.1          39.8       1.0X
      
      Array with 100000 rows:                  Best/Avg Time(ms)    Rate(M/s)   Per Row(ns)   Relative
      ------------------------------------------------------------------------------------------------
      ArrayBuffer                                   5949 / 6028         17.2          58.1       1.0X
      ExternalAppendOnlyUnsafeRowArray              6078 / 6138         16.8          59.4       1.0X
      ```
      
      [B] Comparing `ExternalAppendOnlyUnsafeRowArray` against raw `UnsafeExternalSorter` when there is spilling of data
      
      ```
      Spilling with 1000 rows:                 Best/Avg Time(ms)    Rate(M/s)   Per Row(ns)   Relative
      ------------------------------------------------------------------------------------------------
      UnsafeExternalSorter                          9239 / 9470         28.4          35.2       1.0X
      ExternalAppendOnlyUnsafeRowArray              8857 / 8909         29.6          33.8       1.0X
      
      Spilling with 10000 rows:                Best/Avg Time(ms)    Rate(M/s)   Per Row(ns)   Relative
      ------------------------------------------------------------------------------------------------
      UnsafeExternalSorter                             4 /    5         39.3          25.5       1.0X
      ExternalAppendOnlyUnsafeRowArray                 5 /    6         29.8          33.5       0.8X
      ```
      
      Author: Tejas Patil <tejasp@fb.com>
      
      Closes #16909 from tejasapatil/SPARK-13450_smb_buffer_oom.
      02c274ea
    • hyukjinkwon's avatar
      [SPARK-19872] [PYTHON] Use the correct deserializer for RDD construction for coalesce/repartition · 7387126f
      hyukjinkwon authored
      ## What changes were proposed in this pull request?
      
      This PR proposes to use the correct deserializer, `BatchedSerializer` for RDD construction for coalesce/repartition when the shuffle is enabled. Currently, it is passing `UTF8Deserializer` as is not `BatchedSerializer` from the copied one.
      
      with the file, `text.txt` below:
      
      ```
      a
      b
      
      d
      e
      f
      g
      h
      i
      j
      k
      l
      
      ```
      
      - Before
      
      ```python
      >>> sc.textFile('text.txt').repartition(1).collect()
      ```
      
      ```
      UTF8Deserializer(True)
      Traceback (most recent call last):
        File "<stdin>", line 1, in <module>
        File ".../spark/python/pyspark/rdd.py", line 811, in collect
          return list(_load_from_socket(port, self._jrdd_deserializer))
        File ".../spark/python/pyspark/serializers.py", line 549, in load_stream
          yield self.loads(stream)
        File ".../spark/python/pyspark/serializers.py", line 544, in loads
          return s.decode("utf-8") if self.use_unicode else s
        File "/System/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/encodings/utf_8.py", line 16, in decode
          return codecs.utf_8_decode(input, errors, True)
      UnicodeDecodeError: 'utf8' codec can't decode byte 0x80 in position 0: invalid start byte
      ```
      
      - After
      
      ```python
      >>> sc.textFile('text.txt').repartition(1).collect()
      ```
      
      ```
      [u'a', u'b', u'', u'd', u'e', u'f', u'g', u'h', u'i', u'j', u'k', u'l', u'']
      ```
      
      ## How was this patch tested?
      
      Unit test in `python/pyspark/tests.py`.
      
      Author: hyukjinkwon <gurwls223@gmail.com>
      
      Closes #17282 from HyukjinKwon/SPARK-19872.
      7387126f
    • Herman van Hovell's avatar
      [SPARK-19889][SQL] Make TaskContext callbacks thread safe · 9ff85be3
      Herman van Hovell authored
      ## What changes were proposed in this pull request?
      It is sometimes useful to use multiple threads in a task to parallelize tasks. These threads might register some completion/failure listeners to clean up when the task completes or fails. We currently cannot register such a callback and be sure that it will get called, because the context might be in the process of invoking its callbacks, when the the callback gets registered.
      
      This PR improves this by making sure that you cannot add a completion/failure listener from a different thread when the context is being marked as completed/failed in another thread. This is done by synchronizing these methods on the task context itself.
      
      Failure listeners were called only once. Completion listeners now follow the same pattern; this lifts the idempotency requirement for completion listeners and makes it easier to implement them. In some cases we can (accidentally) add a completion/failure listener after the fact, these listeners will be called immediately in order make sure we can safely clean-up after a task.
      
      As a result of this change we could make the `failure` and `completed` flags non-volatile. The `isCompleted()` method now uses synchronization to ensure that updates are visible across threads.
      
      ## How was this patch tested?
      Adding tests to `TaskContestSuite` to test adding listeners to a completed/failed context.
      
      Author: Herman van Hovell <hvanhovell@databricks.com>
      
      Closes #17244 from hvanhovell/SPARK-19889.
      9ff85be3
    • jiangxingbo's avatar
      [SPARK-19877][SQL] Restrict the nested level of a view · ee36bc1c
      jiangxingbo authored
      ## What changes were proposed in this pull request?
      
      We should restrict the nested level of a view, to avoid stack overflow exception during the view resolution.
      
      ## How was this patch tested?
      
      Add new test case in `SQLViewSuite`.
      
      Author: jiangxingbo <jiangxb1987@gmail.com>
      
      Closes #17241 from jiangxb1987/view-depth.
      ee36bc1c
    • Liwei Lin's avatar
      [SPARK-19817][SS] Make it clear that `timeZone` is a general option in DataStreamReader/Writer · e1ac5534
      Liwei Lin authored
      ## What changes were proposed in this pull request?
      
      As timezone setting can also affect partition values, it works for all formats, we should make it clear.
      
      ## How was this patch tested?
      
      N/A
      
      Author: Liwei Lin <lwlin7@gmail.com>
      
      Closes #17299 from lw-lin/timezone.
      e1ac5534
  5. Mar 14, 2017
    • Xiao Li's avatar
      [SPARK-18112][SQL] Support reading data from Hive 2.1 metastore · f9a93b1b
      Xiao Li authored
      ### What changes were proposed in this pull request?
      This PR is to support reading data from Hive 2.1 metastore. Need to update shim class because of the Hive API changes caused by the following three Hive JIRAs:
      - [HIVE-12730 MetadataUpdater: provide a mechanism to edit the basic statistics of a table (or a partition)](https://issues.apache.org/jira/browse/HIVE-12730)
      - [Hive-13341 Stats state is not captured correctly: differentiate load table and create table](https://issues.apache.org/jira/browse/HIVE-13341)
      - [HIVE-13622 WriteSet tracking optimizations](https://issues.apache.org/jira/browse/HIVE-13622)
      
      There are three new fields added to Hive APIs.
      - `boolean hasFollowingStatsTask`. We always set it to `false`. This is to keep the existing behavior unchanged (starting from 0.13), no matter which Hive metastore client version users choose. If we set it to `true`, the basic table statistics is not collected by Hive. For example,
      
      ```SQL
      	CREATE TABLE tbl AS SELECT 1 AS a
      ```
      When setting `hasFollowingStatsTask ` to `false`, the table properties is like
      ```
      	Properties: [numFiles=1, transient_lastDdlTime=1489513927, totalSize=2]
      ```
      When setting `hasFollowingStatsTask ` to `true`, the table properties is like
      ```
      	Properties: [transient_lastDdlTime=1489513563]
      ```
      
      - `AcidUtils.Operation operation`. Obviously, we do not support ACID. Thus, we set it to `AcidUtils.Operation.NOT_ACID`.
      - `EnvironmentContext environmentContext`. So far, this is always set to `null`. This was introduced for supporting DDL `alter table s update statistics set ('numRows'='NaN')`. Using this DDL, users can specify the statistics. So far, our Spark SQL does not need it, because we use different table properties to store our generated statistics values. However, when Spark SQL issues ALTER TABLE DDL statements, Hive metastore always automatically invalidate the Hive-generated statistics.
      
      In the follow-up PR, we can fix it by explicitly adding a property to `environmentContext`.
      ```JAVA
      putToProperties(StatsSetupConst.STATS_GENERATED, StatsSetupConst.USER)
      ```
      Another alternative is to set `DO_NOT_UPDATE_STATS`to `TRUE`. See the Hive JIRA: https://issues.apache.org/jira/browse/HIVE-15653. We will not address it in this PR.
      
      ### How was this patch tested?
      Added test cases to VersionsSuite.scala
      
      Author: Xiao Li <gatorsmile@gmail.com>
      
      Closes #17232 from gatorsmile/Hive21.
      f9a93b1b
    • hyukjinkwon's avatar
      [SPARK-19828][R] Support array type in from_json in R · d1f6c64c
      hyukjinkwon authored
      ## What changes were proposed in this pull request?
      
      Since we could not directly define the array type in R, this PR proposes to support array types in R as string types that are used in `structField` as below:
      
      ```R
      jsonArr <- "[{\"name\":\"Bob\"}, {\"name\":\"Alice\"}]"
      df <- as.DataFrame(list(list("people" = jsonArr)))
      collect(select(df, alias(from_json(df$people, "array<struct<name:string>>"), "arrcol")))
      ```
      
      prints
      
      ```R
            arrcol
      1 Bob, Alice
      ```
      
      ## How was this patch tested?
      
      Unit tests in `test_sparkSQL.R`.
      
      Author: hyukjinkwon <gurwls223@gmail.com>
      
      Closes #17178 from HyukjinKwon/SPARK-19828.
      d1f6c64c
    • hyukjinkwon's avatar
      [SPARK-19918][SQL] Use TextFileFormat in implementation of TextInputJsonDataSource · 8fb2a02e
      hyukjinkwon authored
      ## What changes were proposed in this pull request?
      
      This PR proposes to use text datasource when Json schema inference.
      
      This basically proposes the similar approach in https://github.com/apache/spark/pull/15813 If we use Dataset for initial loading when inferring the schema, there are advantages. Please refer SPARK-18362
      
      It seems JSON one was supposed to be fixed together but taken out according to https://github.com/apache/spark/pull/15813
      
      > A similar problem also affects the JSON file format and this patch originally fixed that as well, but I've decided to split that change into a separate patch so as not to conflict with changes in another JSON PR.
      
      Also, this seems affecting some functionalities because it does not use `FileScanRDD`. This problem is described in SPARK-19885 (but it was CSV's case).
      
      ## How was this patch tested?
      
      Existing tests should cover this and manual test by `spark.read.json(path)` and check the UI.
      
      Author: hyukjinkwon <gurwls223@gmail.com>
      
      Closes #17255 from HyukjinKwon/json-filescanrdd.
      8fb2a02e
    • Wenchen Fan's avatar
      [SPARK-19887][SQL] dynamic partition keys can be null or empty string · dacc382f
      Wenchen Fan authored
      ## What changes were proposed in this pull request?
      
      When dynamic partition value is null or empty string, we should write the data to a directory like `a=__HIVE_DEFAULT_PARTITION__`, when we read the data back, we should respect this special directory name and treat it as null.
      
      This is the same behavior of impala, see https://issues.apache.org/jira/browse/IMPALA-252
      
      ## How was this patch tested?
      
      new regression test
      
      Author: Wenchen Fan <wenchen@databricks.com>
      
      Closes #17277 from cloud-fan/partition.
      dacc382f
    • Takuya UESHIN's avatar
      [SPARK-19817][SQL] Make it clear that `timeZone` option is a general option in... · 7ded39c2
      Takuya UESHIN authored
      [SPARK-19817][SQL] Make it clear that `timeZone` option is a general option in DataFrameReader/Writer.
      
      ## What changes were proposed in this pull request?
      
      As timezone setting can also affect partition values, it works for all formats, we should make it clear.
      
      ## How was this patch tested?
      
      Existing tests.
      
      Author: Takuya UESHIN <ueshin@databricks.com>
      
      Closes #17281 from ueshin/issues/SPARK-19817.
      7ded39c2
    • Nattavut Sutyanyong's avatar
      [SPARK-18966][SQL] NOT IN subquery with correlated expressions may return incorrect result · 6eac9682
      Nattavut Sutyanyong authored
      ## What changes were proposed in this pull request?
      
      This PR fixes the following problem:
      ````
      Seq((1, 2)).toDF("a1", "a2").createOrReplaceTempView("a")
      Seq[(java.lang.Integer, java.lang.Integer)]((1, null)).toDF("b1", "b2").createOrReplaceTempView("b")
      
      // The expected result is 1 row of (1,2) as shown in the next statement.
      sql("select * from a where a1 not in (select b1 from b where b2 = a2)").show
      +---+---+
      | a1| a2|
      +---+---+
      +---+---+
      
      sql("select * from a where a1 not in (select b1 from b where b2 = 2)").show
      +---+---+
      | a1| a2|
      +---+---+
      |  1|  2|
      +---+---+
      ````
      There are a number of scenarios to consider:
      
      1. When the correlated predicate yields a match (i.e., B.B2 = A.A2)
      1.1. When the NOT IN expression yields a match (i.e., A.A1 = B.B1)
      1.2. When the NOT IN expression yields no match (i.e., A.A1 = B.B1 returns false)
      1.3. When A.A1 is null
      1.4. When B.B1 is null
      1.4.1. When A.A1 is not null
      1.4.2. When A.A1 is null
      
      2. When the correlated predicate yields no match (i.e.,B.B2 = A.A2 is false or unknown)
      2.1. When B.B2 is null and A.A2 is null
      2.2. When B.B2 is null and A.A2 is not null
      2.3. When the value of A.A2 does not match any of B.B2
      
      ````
       A.A1   A.A2      B.B1   B.B2
      -----  -----     -----  -----
          1      1         1      1    (1.1)
          2      1                     (1.2)
       null      1                     (1.3)
      
          1      3      null      3    (1.4.1)
       null      3                     (1.4.2)
      
          1   null         1   null    (2.1)
       null      2                     (2.2 & 2.3)
      ````
      
      We can divide the evaluation of the above correlated NOT IN subquery into 2 groups:-
      
      Group 1: The rows in A when there is a match from the correlated predicate (A.A1 = B.B1)
      
      In this case, the result of the subquery is not empty and the semantics of the NOT IN depends solely on the evaluation of the equality comparison of the columns of NOT IN, i.e., A1 = B1, which says
      
      - If A.A1 is null, the row is filtered (1.3 and 1.4.2)
      - If A.A1 = B.B1, the row is filtered (1.1)
      - If B.B1 is null, any rows of A in the same group (A.A2 = B.B2) is filtered (1.4.1 & 1.4.2)
      - Otherwise, the row is qualified.
      
      Hence, in this group, the result is the row from (1.2).
      
      Group 2: The rows in A when there is no match from the correlated predicate (A.A2 = B.B2)
      
      In this case, all the rows in A, including the rows where A.A1, are qualified because the subquery returns an empty set and by the semantics of the NOT IN, all rows from the parent side qualifies as the result set, that is, the rows from (2.1, 2.2 and 2.3).
      
      In conclusion, the correct result set of the above query is
      ````
       A.A1   A.A2
      -----  -----
          2      1    (1.2)
          1   null    (2.1)
       null      2    (2.2 & 2.3)
      ````
      ## How was this patch tested?
      unit tests, regression tests, and new test cases focusing on the problem being fixed.
      
      Author: Nattavut Sutyanyong <nsy.can@gmail.com>
      
      Closes #17294 from nsyca/18966.
      6eac9682
    • Herman van Hovell's avatar
      [SPARK-19933][SQL] Do not change output of a subquery · e04c05cf
      Herman van Hovell authored
      ## What changes were proposed in this pull request?
      The `RemoveRedundantAlias` rule can change the output attributes (the expression id's to be precise) of a query by eliminating the redundant alias producing them. This is no problem for a regular query, but can cause problems for correlated subqueries: The attributes produced by the subquery are used in the parent plan; changing them will break the parent plan.
      
      This PR fixes this by wrapping a subquery in a `Subquery` top level node when it gets optimized. The `RemoveRedundantAlias` rule now recognizes `Subquery` and makes sure that the output attributes of the `Subquery` node are retained.
      
      ## How was this patch tested?
      Added a test case to `RemoveRedundantAliasAndProjectSuite` and added a regression test to `SubquerySuite`.
      
      Author: Herman van Hovell <hvanhovell@databricks.com>
      
      Closes #17278 from hvanhovell/SPARK-19933.
      e04c05cf
    • Takeshi Yamamuro's avatar
      [SPARK-19923][SQL] Remove unnecessary type conversions per call in Hive · 6325a2f8
      Takeshi Yamamuro authored
      ## What changes were proposed in this pull request?
      This pr removed unnecessary type conversions per call in Hive: https://github.com/apache/spark/blob/master/sql/hive/src/main/scala/org/apache/spark/sql/hive/hiveUDFs.scala#L116
      
      ## How was this patch tested?
      Existing tests
      
      Author: Takeshi Yamamuro <yamamuro@apache.org>
      
      Closes #17264 from maropu/SPARK-19923.
      6325a2f8
    • jiangxingbo's avatar
      [SPARK-18961][SQL] Support `SHOW TABLE EXTENDED ... PARTITION` statement · a02a0b17
      jiangxingbo authored
      ## What changes were proposed in this pull request?
      
      We should support the statement `SHOW TABLE EXTENDED LIKE 'table_identifier' PARTITION(partition_spec)`, just like that HIVE does.
      When partition is specified, the `SHOW TABLE EXTENDED` command should output the information of the partitions instead of the tables.
      Note that in this statement, we require exact matched partition spec. For example:
      ```
      CREATE TABLE show_t1(a String, b Int) PARTITIONED BY (c String, d String);
      ALTER TABLE show_t1 ADD PARTITION (c='Us', d=1) PARTITION (c='Us', d=22);
      
      -- Output the extended information of Partition(c='Us', d=1)
      SHOW TABLE EXTENDED LIKE 'show_t1' PARTITION(c='Us', d=1);
      -- Throw an AnalysisException
      SHOW TABLE EXTENDED LIKE 'show_t1' PARTITION(c='Us');
      ```
      
      ## How was this patch tested?
      Add new test sqls in file `show-tables.sql`.
      Add new test case in `DDLSuite`.
      
      Author: jiangxingbo <jiangxb1987@gmail.com>
      
      Closes #16373 from jiangxb1987/show-partition-extended.
      a02a0b17
    • Menglong TAN's avatar
      [SPARK-11569][ML] Fix StringIndexer to handle null value properly · 85941ecf
      Menglong TAN authored
      ## What changes were proposed in this pull request?
      
      This PR is to enhance StringIndexer with NULL values handling.
      
      Before the PR, StringIndexer will throw an exception when encounters NULL values.
      With this PR:
      - handleInvalid=error: Throw an exception as before
      - handleInvalid=skip: Skip null values as well as unseen labels
      - handleInvalid=keep: Give null values an additional index as well as unseen labels
      
      BTW, I noticed someone was trying to solve the same problem ( #9920 ) but seems getting no progress or response for a long time. Would you mind to give me a chance to solve it ? I'm eager to help. :-)
      
      ## How was this patch tested?
      
      new unit tests
      
      Author: Menglong TAN <tanmenglong@renrenche.com>
      Author: Menglong TAN <tanmenglong@gmail.com>
      
      Closes #17233 from crackcell/11569_StringIndexer_NULL.
      85941ecf
    • zero323's avatar
      [SPARK-19940][ML][MINOR] FPGrowthModel.transform should skip duplicated items · d4a637cd
      zero323 authored
      ## What changes were proposed in this pull request?
      
      This commit moved `distinct` in its intended place to avoid duplicated predictions and adds unit test covering the issue.
      
      ## How was this patch tested?
      
      Unit tests.
      
      Author: zero323 <zero323@users.noreply.github.com>
      
      Closes #17283 from zero323/SPARK-19940.
      d4a637cd
Loading