Making paths to fasterai files relative

fasterai/callbacks.py

@@ -1,9 +1,9 @@
 from fastai.core import *
 from fastai.sgdr import Callback
 from fastai.dataset import ModelData, ImageData
-from fasterai.visualize import ModelStatsVisualizer, ImageGenVisualizer, GANTrainerStatsVisualizer
-from fasterai.visualize import LearnerStatsVisualizer, ModelGraphVisualizer, ModelHistogramVisualizer
-from fasterai.training import GenResult, CriticResult, GANTrainer
+from .visualize import ModelStatsVisualizer, ImageGenVisualizer, GANTrainerStatsVisualizer
+from .visualize import LearnerStatsVisualizer, ModelGraphVisualizer, ModelHistogramVisualizer
+from .training import GenResult, CriticResult, GANTrainer
 from tensorboardX import SummaryWriter
 
 def clear_directory(dir:Path):

fasterai/generators.py

@@ -1,6 +1,6 @@
 from fastai.core import *
 from fastai.conv_learner import model_meta, cut_model
-from fasterai.modules import ConvBlock, UnetBlock, UpSampleBlock, SaveFeatures
+from .modules import ConvBlock, UnetBlock, UpSampleBlock, SaveFeatures
 from abc import ABC, abstractmethod
 
 class GeneratorModule(ABC, nn.Module):

fasterai/loss.py

@@ -2,7 +2,7 @@
 from fastai.torch_imports import *
 from fastai.core import *
 from fastai.conv_learner import children
-from fasterai.modules import SaveFeatures
+from .modules import SaveFeatures
 import torchvision.models as models
 
 

fasterai/training.py

@@ -3,9 +3,9 @@ from fastai.torch_imports import *
 from fastai.dataset import Transform
 from fastai.layer_optimizer import LayerOptimizer
 from fastai.sgdr import CircularLR_beta
-from fasterai.modules import ConvBlock
-from fasterai.generators import GeneratorModule
-from fasterai.dataset import ImageGenDataLoader
+from .modules import ConvBlock
+from .generators import GeneratorModule
+from .dataset import ImageGenDataLoader
 from collections import Iterable
 import torch.utils.hooks as hooks
 from abc import ABC

fasterai/visualize.py

@@ -5,8 +5,8 @@ from matplotlib.axes import Axes
 from fastai.dataset import FilesDataset, ImageData, ModelData, open_image
 from fastai.transforms import Transform, scale_min, tfms_from_stats, inception_stats
 from fastai.transforms import CropType, NoCrop, Denormalize
-from fasterai.training import GenResult, CriticResult, GANTrainer
-from fasterai.images import ModelImageSet, EasyTensorImage
+from .training import GenResult, CriticResult, GANTrainer
+from .images import ModelImageSet, EasyTensorImage
 from IPython.display import display
 from tensorboardX import SummaryWriter
 import torchvision.utils as vutils