[update] 20250309

Author: Lmy0217

.travis.yml

@@ -4,6 +4,8 @@ python:
   - "3.8"
   - "3.9"
   - "3.10"
+  - "3.11"
+  - "3.12"
 
 install:
   - pip install -r requirements.txt

build.py

@@ -1,9 +1,9 @@
 import os
 import shutil
 import sys
-from distutils.core import Extension, setup
 
 from Cython.Build import cythonize
+from setuptools import Extension, setup
 
 
 def walk(root, exclude=(), copy=()):

circle.yml

@@ -3,14 +3,16 @@ version: 2.1
 workflows:
   test:
     jobs:
-      - test-py38
-      - test-py39
-      - test-py310
+      - job-py38
+      - job-py39
+      - job-py310
+      - job-py311
+      - job-py312
 
 jobs:
-  test-py36: &test-template
+  job-py38: &job-template
     docker:
-      - image: circleci/python:3.6
+      - image: cimg/python:3.8
 
     working_directory: ~/repo
 
@@ -59,17 +61,22 @@ jobs:
           path: test-reports
           destination: test-reports
 
-  test-py38:
-    <<: *test-template
+  job-py39:
+    <<: *job-template
     docker:
-      - image: circleci/python:3.8
+      - image: cimg/python:3.9
 
-  test-py39:
-    <<: *test-template
+  job-py310:
+    <<: *job-template
     docker:
-      - image: circleci/python:3.9
+      - image: cimg/python:3.10
 
-  test-py310:
-    <<: *test-template
+  job-py311:
+    <<: *job-template
     docker:
-      - image: circleci/python:3.10
+      - image: cimg/python:3.11
+
+  job-py312:
+    <<: *job-template
+    docker:
+      - image: cimg/python:3.12

configs/Env.py

@@ -19,6 +19,12 @@ def __init__(self):
                 os.path.join(os.path.dirname(__file__), '..', self.paths.root_folder))
         else:
             raise ValueError('Lack of `res/env/paths.json` file or `root_folder` value')
+        self.set_environ()
+
+    def set_environ(self):
+        if hasattr(self, 'environ'):
+            for key, value in self.environ.dict().items():
+                os.environ[key] = value
 
     def getdir(self, path):
         return os.path.abspath(os.path.join(self.paths.root_folder, path))

res/env/environ.json

@@ -0,0 +1,3 @@
+{
+  "HF_ENDPOINT": "https://hf-mirror.com"
+}

utils/common.py

@@ -15,6 +15,7 @@
 
 
 def set_seed(seed=0):
+    seed = int(seed)
     torch.manual_seed(seed)
     torch.cuda.manual_seed(seed)
     torch.cuda.manual_seed_all(seed)