fix: fix tests

unit-mesh · Dec 19, 2023 · 6ce7891 · 6ce7891
1 parent ebcd6d4
commit 6ce7891
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/finetunes/deepseek/finetune.ipynb b/finetunes/deepseek/finetune.ipynb
@@ -46,11 +46,10 @@
    "source": [
     "# Make sure you have git-lfs installed (https://git-lfs.com)\n",
     "!git lfs install\n",
-    "!git clone https://huggingface.co/deepseek-ai/deepseek-coder-6.7b-instruct\n",
-    "\n",
     "# if you want to clone without large files – just their pointers\n",
     "# prepend your git clone with the following env var:\n",
-    "!GIT_LFS_SKIP_SMUDGE=1"
+    "!GIT_LFS_SKIP_SMUDGE=1\n",
+    "!git clone https://huggingface.co/deepseek-ai/deepseek-coder-6.7b-instruct"
    ]
   },
   {

diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/ext/CodeDataStructUtil.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/ext/CodeDataStructUtil.kt
@@ -6,9 +6,13 @@ import chapi.domain.core.CodePosition
 fun CodeDataStruct.toUml(): String {
     val output = StringBuilder()
 
-    val superClass = Implements + Extend
+    val superClass = Implements + if (Extend.isNotBlank()) {
+        listOf(Extend)
+    } else {
+        emptyList()
+    }
     val superClasses = if (superClass.isNotEmpty()) {
-        ":" + superClass.joinToString(", ") + " "
+        ": " + superClass.joinToString(", ") + " "
     } else {
         ""
     }

diff --git a/...picker/src/test/kotlin/cc/unitmesh/pick/prompt/strategy/RelatedCodeStrategyBuilderTest.kt b/...picker/src/test/kotlin/cc/unitmesh/pick/prompt/strategy/RelatedCodeStrategyBuilderTest.kt
@@ -141,7 +141,7 @@ public class HelloController {
 // 
 //  }
 // 
-// class BlogRepository {
+// class BlogRepository : CrudRepository<BlogPost,Long> {
 // 
 //  }
 // """
-Original file line number
+Diff line change
@@ Expand Up / @@ -141,7 +141,7 @@ public class HelloController { @@
     //
     //  }
     //
-    // class BlogRepository {
+    // class BlogRepository : CrudRepository<BlogPost,Long> {
     //
     //  }
     // """
@@ Expand Down @@