Small refactoring

hyperskill · Sep 18, 2023 · 12e6fdd · 12e6fdd
1 parent fd52c60
commit 12e6fdd
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/...main/kotlin/org/jetbrains/research/code/submissions/clustering/impl/unifiers/PyUnifier.kt b/...main/kotlin/org/jetbrains/research/code/submissions/clustering/impl/unifiers/PyUnifier.kt
@@ -27,10 +27,12 @@ class PyUnifier(
     transformationsConfig: TransformationsConfig = defaultTransformationsConfig
 ) : AbstractUnifier(project) {
     override val language = Language.PYTHON
-    override val repeatingTransformations = transformationsConfig.repeatingTransformations.map {
-        PyTransformations.valueOf(it).transformation
-    }
-    override val singleRunTransformations: List<Transformation> = transformationsConfig.singleTransformations.map {
+    override val repeatingTransformations = transformationsConfig
+        .repeatingTransformations.toPyTransformations()
+    override val singleRunTransformations: List<Transformation> = transformationsConfig
+        .singleTransformations.toPyTransformations()
+
+    private fun List<String>.toPyTransformations() = map {
         PyTransformations.valueOf(it).transformation
     }
 
@@ -55,8 +57,9 @@ class PyUnifier(
 
     companion object {
         val defaultTransformationsConfig = TransformationsConfig(
+            // TODO: update Kotlin and replace to entities
             repeatingTransformations = PyTransformations.values().map { it.name },
-            singleTransformations = listOf()
+            singleTransformations = emptyList()
         )
     }
 }