chore: 更新项目名称

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