Skip to content

Commit a3b4736

Browse files
committed
fix nits
1 parent 16ad74a commit a3b4736

File tree

4 files changed

+11
-19
lines changed

4 files changed

+11
-19
lines changed

src/main/scala/ldbc/snb/datagen/model/package.scala

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -60,16 +60,10 @@ package object model {
6060
cardinality: Cardinality,
6161
isStatic: Boolean = false,
6262
sourceNameOverride: Option[String] = None,
63-
destinationNameOverride: Option[String] = None,
63+
destinationNameOverride: Option[String] = None
6464
) extends EntityType {
65-
val sourceName: String = sourceNameOverride match {
66-
case Some(name) => name
67-
case _ => source.name
68-
}
69-
val destinationName: String = destinationNameOverride match {
70-
case Some(name) => name
71-
case _ => destination.name
72-
}
65+
val sourceName: String = sourceNameOverride.getOrElse(source.name)
66+
val destinationName: String = destinationNameOverride.getOrElse(destination.name)
7367

7468
override val entityPath: String = s"${s(isStatic)}/${sourceName}_${pascalToCamel(`type`)}_${destinationName}"
7569

src/main/scala/ldbc/snb/datagen/transformation/transform/ExplodeEdges.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
package ldbc.snb.datagen.transformation.transform
22

3-
import ldbc.snb.datagen.model.Cardinality.{NN, NOne, OneN}
4-
import ldbc.snb.datagen.model.EntityType.{Edge, Node}
3+
import ldbc.snb.datagen.model.Cardinality._
4+
import ldbc.snb.datagen.model.EntityType._
55
import ldbc.snb.datagen.model.Mode
66
import ldbc.snb.datagen.model.Mode.Raw.withRawColumns
77
import ldbc.snb.datagen.model.raw._

src/main/scala/ldbc/snb/datagen/transformation/transform/RawToBiTransform.scala

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,9 @@
11
package ldbc.snb.datagen.transformation.transform
22

3-
import ldbc.snb.datagen.model.Cardinality.{NOne, OneN}
4-
import ldbc.snb.datagen.model.EntityType.Edge
3+
import ldbc.snb.datagen.model.Cardinality._
4+
import ldbc.snb.datagen.model.EntityType._
55
import ldbc.snb.datagen.model.Mode.BI
66
import ldbc.snb.datagen.model._
7-
import ldbc.snb.datagen.model.graphs.Raw
8-
import ldbc.snb.datagen.model.raw._
97
import ldbc.snb.datagen.syntax._
108
import ldbc.snb.datagen.util.Logging
119
import ldbc.snb.datagen.util.sql._

src/main/scala/ldbc/snb/datagen/transformation/transform/RawToInteractiveTransform.scala

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,14 @@
11
package ldbc.snb.datagen.transformation.transform
22

3-
import ldbc.snb.datagen.model.Cardinality.NN
4-
import ldbc.snb.datagen.model.EntityType.Edge
5-
import ldbc.snb.datagen.model.raw.PersonType
3+
import ldbc.snb.datagen.model.Cardinality._
4+
import ldbc.snb.datagen.model.EntityType._
5+
import ldbc.snb.datagen.model.raw._
66
import ldbc.snb.datagen.model.{EntityType, Graph, Mode}
77
import ldbc.snb.datagen.syntax._
88
import ldbc.snb.datagen.util.Logging
99
import ldbc.snb.datagen.util.sql._
1010
import org.apache.spark.sql.DataFrame
11-
import org.apache.spark.sql.functions.{col, lit, to_timestamp}
11+
import org.apache.spark.sql.functions.{col, lit}
1212

1313
case class RawToInteractiveTransform(mode: Mode.Interactive, simulationStart: Long, simulationEnd: Long)
1414
extends Transform[Mode.Raw.type, Mode.Interactive]

0 commit comments

Comments
 (0)