Skip to content

Commit c922e54

Browse files
committed
Fix ktlint-detected issues
1 parent 2c5db7e commit c922e54

File tree

8 files changed

+14
-62
lines changed

8 files changed

+14
-62
lines changed

android/src/main/java/com/swmansion/rnexecutorch/SpeechToText.kt

+2-2
Original file line numberDiff line numberDiff line change
@@ -57,10 +57,10 @@ class SpeechToText(
5757
promise: Promise,
5858
) {
5959
val encoding = writableArrayToEValue(this.speechToTextModule.encode(waveform))
60-
val generatedTokens = mutableListOf(this.speechToTextModule.START_TOKEN)
60+
val generatedTokens = mutableListOf(this.speechToTextModule.startToken)
6161
var lastToken = 0
6262
Thread {
63-
while (lastToken != this.speechToTextModule.EOS_TOKEN) {
63+
while (lastToken != this.speechToTextModule.eosToken) {
6464
// TODO uncomment, for now
6565
// lastToken = this.speechToTextModule.decode(generatedTokens, encoding)
6666
emitOnToken(lastToken.toDouble())

android/src/main/java/com/swmansion/rnexecutorch/models/StyleTransferModel.kt

+1-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class StyleTransferModel(
2929
override fun postprocess(output: Array<EValue>): Mat {
3030
val tensor = output[0].toTensor()
3131
val modelShape = getModelImageSize()
32-
val result = ImageProcessor.EValueToMat(tensor.dataAsFloatArray, modelShape.width.toInt(), modelShape.height.toInt())
32+
val result = ImageProcessor.eValueToMat(tensor.dataAsFloatArray, modelShape.width.toInt(), modelShape.height.toInt())
3333
Imgproc.resize(result, result, originalSize)
3434
return result
3535
}

android/src/main/java/com/swmansion/rnexecutorch/models/objectDetection/SSDLiteLargeModel.kt

+4-4
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@ import org.opencv.core.Size
1212
import org.opencv.imgproc.Imgproc
1313
import org.pytorch.executorch.EValue
1414

15-
const val detectionScoreThreshold = .7f
16-
const val iouThreshold = .55f
15+
const val DETECTION_SCORE_THRESHOLD = .7f
16+
const val IOU_THRESHOLD = .55f
1717

1818
class SSDLiteLargeModel(
1919
reactApplicationContext: ReactApplicationContext,
@@ -52,7 +52,7 @@ class SSDLiteLargeModel(
5252
val detections: MutableList<Detection> = mutableListOf()
5353
for (idx in 0 until numel.toInt()) {
5454
val score = scores[idx]
55-
if (score < detectionScoreThreshold) {
55+
if (score < DETECTION_SCORE_THRESHOLD) {
5656
continue
5757
}
5858
val bbox =
@@ -68,7 +68,7 @@ class SSDLiteLargeModel(
6868
)
6969
}
7070

71-
val detectionsPostNms = nms(detections, iouThreshold)
71+
val detectionsPostNms = nms(detections, IOU_THRESHOLD)
7272
return detectionsPostNms.toTypedArray()
7373
}
7474
}

android/src/main/java/com/swmansion/rnexecutorch/models/speechToText/BaseS2TModule.kt

+2-2
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@ import com.swmansion.rnexecutorch.models.BaseModel
77
abstract class BaseS2TModule {
88
lateinit var encoder: BaseModel<ReadableArray, WritableArray>
99
lateinit var decoder: BaseS2TDecoder
10-
abstract var START_TOKEN: Int
11-
abstract var EOS_TOKEN: Int
10+
abstract var startToken: Int
11+
abstract var eosToken: Int
1212

1313
fun encode(input: ReadableArray): WritableArray = this.encoder.runModel(input)
1414

android/src/main/java/com/swmansion/rnexecutorch/models/speechToText/Moonshine.kt

+2-2
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@ package com.swmansion.rnexecutorch.models.speechtotext
33
import com.facebook.react.bridge.ReadableArray
44

55
class Moonshine : BaseS2TModule() {
6-
override var START_TOKEN = 1
7-
override var EOS_TOKEN = 2
6+
override var startToken = 1
7+
override var eosToken = 2
88

99
override fun decode(
1010
prevTokens: ReadableArray,

android/src/main/java/com/swmansion/rnexecutorch/models/speechToText/Whisper.kt

+2-2
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@ package com.swmansion.rnexecutorch.models.speechtotext
33
import com.facebook.react.bridge.ReadableArray
44

55
class Whisper : BaseS2TModule() {
6-
override var START_TOKEN = 50257
7-
override var EOS_TOKEN = 50256
6+
override var startToken = 50257
7+
override var eosToken = 50256
88

99
override fun decode(
1010
prevTokens: ReadableArray,

android/src/main/java/com/swmansion/rnexecutorch/utils/ImageProcessor.kt

+1-1
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ class ImageProcessor {
7272
)
7373
}
7474

75-
fun EValueToMat(
75+
fun eValueToMat(
7676
array: FloatArray,
7777
width: Int,
7878
height: Int,

android/src/main/java/com/swmansion/rnexecutorch/utils/ProgressResponseBody.kt

-48
This file was deleted.

0 commit comments

Comments
 (0)