chore: 更新项目名称

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):