edit filename: chattool=>chattype

cubenlp · Mar 10, 2024 · b9c04c6 · b9c04c6
1 parent 0027578
commit b9c04c6
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/chattool/__init__.py b/chattool/__init__.py
@@ -5,7 +5,7 @@
 __version__ = '3.1.1'
 
 import os, sys, requests
-from .chattool import Chat, Resp
+from .chattype import Chat, Resp
 from .checkpoint import load_chats, process_chats
 from .proxy import proxy_on, proxy_off, proxy_status
 from . import request

diff --git a/chattool/chattool.py → chattool/chattype.py b/chattool/chattool.py → chattool/chattype.py
diff --git a/chattool/checkpoint.py b/chattool/checkpoint.py
@@ -1,6 +1,6 @@
 import json, warnings, os
 from typing import List, Dict, Union, Callable, Any
-from .chattool import Chat
+from .chattype import Chat
 import tqdm
 
 def load_chats( checkpoint:str):

diff --git a/tests/test_async.py b/tests/test_async.py
@@ -53,9 +53,8 @@ async def show_resp(chat):
 def test_async_process():
     chkpoint = testpath + "test_async.jsonl"
     t = time.time()
-    resp = async_chat_completion(chatlogs[:1], chkpoint, clearfile=True, nproc=3)
-    resp = async_chat_completion(chatlogs, chkpoint, nproc=3)
-    assert all(resp)
+    async_chat_completion(chatlogs[:1], chkpoint, clearfile=True, nproc=3)
+    async_chat_completion(chatlogs, chkpoint, nproc=3)
     print(f"Time elapsed: {time.time() - t:.2f}s")
 
 # broken test