From 39471b3a0380e2e94901c7c47f180cdba65cbc46 Mon Sep 17 00:00:00 2001 From: laynholt Date: Tue, 25 Nov 2025 14:06:56 +0000 Subject: [PATCH] modelV renamed to mediarV --- core/models/__init__.py | 12 ++++++------ core/models/{model_v.py => mediar_v.py} | 14 +++++++------- 2 files changed, 13 insertions(+), 13 deletions(-) rename core/models/{model_v.py => mediar_v.py} (95%) diff --git a/core/models/__init__.py b/core/models/__init__.py index f29508f..6766d91 100644 --- a/core/models/__init__.py +++ b/core/models/__init__.py @@ -2,13 +2,13 @@ from torch import nn from typing import Final, Type, Any from pydantic import BaseModel -from .model_v import ModelV, ModelVParams +from .mediar_v import MediarV, MediarVParams __all__ = [ "ModelRegistry", - "ModelV", - "ModelVParams" + "MediarV", + "MediarVParams" ] @@ -17,9 +17,9 @@ class ModelRegistry: # Single dictionary storing both model classes and parameter classes. __MODELS: Final[dict[str, dict[str, Type[Any]]]] = { - "ModelV": { - "class": ModelV, - "params": ModelVParams, + "MediarV": { + "class": MediarV, + "params": MediarVParams, }, } diff --git a/core/models/model_v.py b/core/models/mediar_v.py similarity index 95% rename from core/models/model_v.py rename to core/models/mediar_v.py index 8bd109d..eceb8e5 100644 --- a/core/models/model_v.py +++ b/core/models/mediar_v.py @@ -7,10 +7,10 @@ from segmentation_models_pytorch.base.modules import Activation from pydantic import BaseModel, ConfigDict -__all__ = ["ModelV"] +__all__ = ["MediarV"] -class ModelVParams(BaseModel): +class MediarVParams(BaseModel): model_config = ConfigDict(frozen=True) encoder_name: str = "mit_b5" # Default encoder @@ -26,19 +26,19 @@ class ModelVParams(BaseModel): def asdict(self) -> dict[str, Any]: """ - Returns a dictionary of valid parameters for `nn.ModelV`. + Returns a dictionary of valid parameters for `nn.MediarV`. Returns: - dict(str, Any): Dictionary of parameters for nn.ModelV. + dict(str, Any): Dictionary of parameters for nn.MediarV. """ loss_kwargs = self.model_dump() return {k: v for k, v in loss_kwargs.items() if v is not None} # Remove None values -class ModelV(MAnet): - """ModelV model""" +class MediarV(MAnet): + """MediarV model""" - def __init__(self, params: ModelVParams) -> None: + def __init__(self, params: MediarVParams) -> None: # Initialize the MAnet model with provided parameters super().__init__(**params.asdict())