From 56fff47001a7ea728b8cb76612bfbc7d2dd295b5 Mon Sep 17 00:00:00 2001 From: lucasdelimanogueira Date: Sat, 25 May 2024 09:55:02 -0300 Subject: [PATCH] fix import data module --- norch/__init__.py | 1 + norch/utils/data/__init__.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/norch/__init__.py b/norch/__init__.py index c2a3122..80da58b 100644 --- a/norch/__init__.py +++ b/norch/__init__.py @@ -3,6 +3,7 @@ from .nn import * from .optim import * from .utils import * from .norchvision import * +from . import data __version__ = "0.0.4" __author__ = 'Lucas de Lima Nogueira' diff --git a/norch/utils/data/__init__.py b/norch/utils/data/__init__.py index 2397554..e5ea955 100644 --- a/norch/utils/data/__init__.py +++ b/norch/utils/data/__init__.py @@ -1,4 +1,5 @@ from .dataset import * from .example import * from .dataloader import * -from .batch import * \ No newline at end of file +from .batch import * +from .distributed import * \ No newline at end of file