diff --git a/src/mflux/config/model_config.py b/src/mflux/config/model_config.py index f8ced85..46e30e6 100644 --- a/src/mflux/config/model_config.py +++ b/src/mflux/config/model_config.py @@ -46,7 +46,7 @@ class ModelLookup: @staticmethod def from_alias(alias: str) -> ModelConfig: warnings.warn( - "from_alias is deprecated and will be removed in a future release. " "Please use from_name instead.", + "from_alias is deprecated and will be removed in a future release. Please use from_name instead.", DeprecationWarning, stacklevel=2, ) diff --git a/src/mflux/flux/flux.py b/src/mflux/flux/flux.py index f18514d..f01b541 100644 --- a/src/mflux/flux/flux.py +++ b/src/mflux/flux/flux.py @@ -139,7 +139,7 @@ class Flux1(nn.Module): @staticmethod def from_alias(alias: str, quantize: int | None = None) -> "Flux1": warnings.warn( - "from_alias is deprecated and will be removed in a future release. " "Please use from_name instead.", + "from_alias is deprecated and will be removed in a future release. Please use from_name instead.", DeprecationWarning, stacklevel=2, )