Skip to content

Commit 1b4498b

Browse files
committed
release 0.2.6
1 parent 081085e commit 1b4498b

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

build.sbt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ import Deps._
22

33
name := "ltrlib"
44

5-
version := "0.2.5"
5+
version := "0.2.6"
66

77
scalaVersion := "2.13.14"
88

src/main/scala/io/github/metarank/ltrlib/booster/Booster.scala

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,14 @@ import io.github.metarank.ltrlib.model.{Dataset, Model}
55
import org.apache.commons.math3.linear.{Array2DRowRealMatrix, ArrayRealVector, RealMatrix, RealVector}
66

77
trait Booster[D] extends Model {
8-
protected var isClosed = false
8+
protected var nativeLibIsClosed = false
99
def save(): Array[Byte]
1010
def predictMat(values: Array[Double], rows: Int, cols: Int): Array[Double]
1111
def weights(): Array[Double]
1212
def close(): Unit
13+
def isClosed(): Boolean = nativeLibIsClosed
1314

14-
def whenNotClosed[T](f: => T): T = if (!isClosed) f else throw new Exception("booster is already closed")
15+
def whenNotClosed[T](f: => T): T = if (!nativeLibIsClosed) f else throw new Exception("booster is already closed")
1516

1617
override def predict(values: RealMatrix): ArrayRealVector = {
1718
val rows = values.getRowDimension

src/main/scala/io/github/metarank/ltrlib/booster/LightGBMBooster.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ case class LightGBMBooster(model: LGBMBooster) extends Booster[LGBMDataset] with
1616
}
1717

1818
override def close(): Unit = whenNotClosed {
19-
isClosed = true
19+
nativeLibIsClosed = true
2020
model.close()
2121
}
2222

0 commit comments

Comments
 (0)