style: rename

src/domain/model/__init__.py

@@ -1,7 +1,7 @@
-from ._autoencoder import AutoencoderModel
-from ._pipeline import PipelineModel
+from ._autoencoder import Autoencoder
+from ._pipeline import Pipeline
 
 __all__ = [
-    "PipelineModel",
-    "AutoencoderModel",
+    "Pipeline",
+    "Autoencoder",
 ]

src/domain/model/_autoencoder.py

@@ -4,6 +4,6 @@ from diffusers import AutoencoderKL
 
 
 @dataclass
-class AutoencoderModel:
+class Autoencoder:
     autoencoder: AutoencoderKL
     path: str

src/domain/model/_pipeline.py

@@ -4,6 +4,6 @@ from diffusers import StableDiffusionPipeline
 
 
 @dataclass
-class PipelineModel:
+class Pipeline:
     pipeline: StableDiffusionPipeline
     path: str

src/domain/repositories/__init__.py

@@ -1,4 +1,4 @@
-from ._i_autoencoder_repository import IAutoencoderRepository
-from ._i_pipeline_repository import IPipelineRepository
+from ._autoencoder_repository import AutoencoderRepository
+from ._pipeline_repository import PipelineRepository
 
-__all__ = ["IPipelineRepository", "IAutoencoderRepository"]
+__all__ = ["PipelineRepository", "AutoencoderRepository"]

src/domain/repositories/_autoencoder_repository.py

@@ -4,7 +4,7 @@ import torch
 from diffusers import AutoencoderKL
 
 
-class IAutoencoderRepository(ABC):
+class AutoencoderRepository(ABC):
     @abstractmethod
     def convert_and_save_from_single_file(self, checkpoint_path: str) -> str:
         pass

src/domain/repositories/_pipeline_repository.py

@@ -4,7 +4,7 @@ import torch
 from diffusers import StableDiffusionPipeline
 
 
-class IPipelineRepository(ABC):
+class PipelineRepository(ABC):
     @abstractmethod
     def convert_and_save_from_single_file(self, checkpoint_path: str, dtype: torch.dtype) -> str:
         pass