Skip to content

Commit 9f9ef47

Browse files
ClarkChin08ftian1
authored andcommitted
[fix] move import data and transform to data
1 parent 4deacb6 commit 9f9ef47

File tree

5 files changed

+4
-6
lines changed

5 files changed

+4
-6
lines changed

lpot/benchmark.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
# limitations under the License.
1717

1818
from .utils import logger
19-
from .experimental.data import DATALOADERS
19+
from .data import DATALOADERS
2020
from .experimental import Benchmark as ExpBenchmark
2121

2222
class Benchmark(object):

lpot/data/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@
1717

1818

1919
from .dataloaders import DataLoader
20+
import lpot.data.datasets
21+
import lpot.data.transforms
2022
from ..experimental.data.datasets import DATASETS, Dataset, IterableDataset, dataset_registry
2123
from ..experimental.data.transforms import TRANSFORMS, BaseTransform, transform_registry
2224
from ..experimental.data.dataloaders import DATALOADERS

lpot/experimental/data/datasets/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,6 @@
1919
from os.path import dirname, basename, isfile, join
2020
import glob
2121

22-
import lpot.data.datasets
23-
2422
modules = glob.glob(join(dirname(__file__), "*.py"))
2523

2624
for f in modules:

lpot/experimental/data/transforms/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,6 @@
1919
from os.path import dirname, basename, isfile, join
2020
import glob
2121

22-
import lpot.data.transforms
23-
2422
modules = glob.glob(join(dirname(__file__), "*.py"))
2523

2624
for f in modules:

lpot/quantization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
import os
1919
from .utils import logger
20-
from .experimental.data import DATALOADERS, DATASETS
20+
from .data import DATALOADERS, DATASETS
2121
from .experimental import Quantization as ExpQuantization
2222

2323
class Quantization(object):

0 commit comments

Comments
 (0)