Fix: Remove __main__ relative path

Github issue #61

.pre-commit-config.yaml

@@ -14,6 +14,6 @@ repos:
     rev: stable
     hooks:
       - id: black
-        exclude:|
+        exclude: |
           src/scratch/htdocs
           .github/workflows/testing.yaml

src/__init__.py

@@ -1,3 +1,3 @@
 from .scratch import main, streamlit_frontend, gpt3_api
-from .stable_diffusion import main, test
+from .stable_diffusion import dataset, train, utils
 from .redis_celery import main, streamlit_frontend, gpt3_api, worker

src/stable_diffusion/__init__.py

@@ -1,4 +1,3 @@
 from .dataset import valid_prompt, MelonDataset
-from .main import main
-from .test import test
 from .train import train, valid
+from .utils import plot, training, util

src/stable_diffusion/train.py

@@ -10,13 +10,13 @@
 
 from diffusers import StableDiffusionPipeline
 
-from .utils.training import compute_snr
-from .utils.plot import make_image_grid
-
 from PIL import Image
 import numpy as np
 from functools import partial
 
+from .utils.training import compute_snr
+from .utils.plot import make_image_grid
+
 
 def train(
     args,