chore: 将data 模块命名为dataset
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
import numpy as np
|
||||
import torch
|
||||
|
||||
from astrai.data.dataset import DatasetFactory
|
||||
from astrai.dataset.dataset import DatasetFactory
|
||||
from astrai.serialization import save_h5
|
||||
|
||||
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
from astrai.data import ResumableDistributedSampler
|
||||
from astrai.dataset import ResumableDistributedSampler
|
||||
|
||||
|
||||
def test_random_sampler_consistency(random_dataset):
|
||||
|
||||
Reference in New Issue
Block a user