chore: 更新项目名称

tests/conftest.py

@@ -7,9 +7,9 @@ import torch
 import pytest
 
 from torch.utils.data import Dataset
-from khaosz.config.model_config import ModelConfig
-from khaosz.data.tokenizer import BpeTokenizer
-from khaosz.model.transformer import Transformer
+from astrai.config.model_config import ModelConfig
+from astrai.data.tokenizer import BpeTokenizer
+from astrai.model.transformer import Transformer
 
 
 class RandomDataset(Dataset):

tests/data/test_checkpoint.py

@@ -4,8 +4,8 @@ import torch.distributed as dist
 
 from torch.optim import AdamW
 from torch.optim.lr_scheduler import CosineAnnealingLR
-from khaosz.data.serialization import Checkpoint
-from khaosz.parallel.setup import get_rank, spawn_parallel_fn
+from astrai.data.serialization import Checkpoint
+from astrai.parallel.setup import get_rank, spawn_parallel_fn
 
 
 def test_single_process():

tests/data/test_dataset.py

@@ -1,8 +1,8 @@
 import torch
 import numpy as np
 
-from khaosz.data.serialization import save_h5
-from khaosz.data.dataset import *
+from astrai.data.serialization import save_h5
+from astrai.data.dataset import *
 
 
 def test_dataset_loader_random_paths(base_test_env):

tests/data/test_sampler.py

@@ -1,5 +1,5 @@
-from khaosz.trainer import *
-from khaosz.data import *
+from astrai.trainer import *
+from astrai.data import *
 
 
 def test_random_sampler_consistency(random_dataset):

tests/module/test_module.py

@@ -5,11 +5,11 @@ import shutil
 import pytest
 import tempfile
 import safetensors.torch as st
-from khaosz.trainer import *
-from khaosz.config import *
-from khaosz.model import *
-from khaosz.data import *
-from khaosz.inference.generator import EmbeddingEncoderCore, GeneratorCore
+from astrai.trainer import *
+from astrai.config import *
+from astrai.model import *
+from astrai.data import *
+from astrai.inference.generator import EmbeddingEncoderCore, GeneratorCore
 from tokenizers import pre_tokenizers
 
 

tests/module/test_tie_weight.py

@@ -4,8 +4,8 @@ import torch
 import pytest
 import tempfile
 import safetensors.torch as st
-from khaosz.model.transformer import Transformer
-from khaosz.config.model_config import ModelConfig
+from astrai.model.transformer import Transformer
+from astrai.config.model_config import ModelConfig
 
 
 @pytest.fixture

tests/test_parallel.py

@@ -1,7 +1,7 @@
 import torch
 import torch.distributed as dist
 
-from khaosz.parallel import get_rank, only_on_rank, spawn_parallel_fn
+from astrai.parallel import get_rank, only_on_rank, spawn_parallel_fn
 
 
 @only_on_rank(0)

tests/trainer/test_callbacks.py

@@ -1,7 +1,7 @@
 import torch
 
-from khaosz.config import *
-from khaosz.trainer import *
+from astrai.config import *
+from astrai.trainer import *
 
 
 def test_callback_integration(base_test_env, random_dataset):

tests/trainer/test_early_stopping.py

@@ -1,9 +1,9 @@
 import os
 import torch
 import numpy as np
-from khaosz.config import *
-from khaosz.trainer import *
-from khaosz.data.serialization import Checkpoint
+from astrai.config import *
+from astrai.trainer import *
+from astrai.data.serialization import Checkpoint
 
 
 def test_early_stopping_simulation(base_test_env, early_stopping_dataset):

tests/trainer/test_train_strategy.py

@@ -2,9 +2,9 @@ import torch
 import numpy as np
 import pytest
 
-from khaosz.config import *
-from khaosz.trainer.schedule import *
-from khaosz.data.dataset import *
+from astrai.config import *
+from astrai.trainer.schedule import *
+from astrai.data.dataset import *
 
 
 def test_schedule_factory_random_configs():

tests/trainer/test_trainer.py

@@ -2,9 +2,9 @@ import torch
 import numpy as np
 
 
-from khaosz.config import *
-from khaosz.trainer import *
-from khaosz.data.dataset import *
+from astrai.config import *
+from astrai.trainer import *
+from astrai.data.dataset import *
 
 
 def test_different_batch_sizes(base_test_env, random_dataset):