Update segmentation_models_pytorch/encoders/__init__.py

qubvel · adamjstewart · web-flow · commit 7bb9d378cd33 · 2025-01-15T12:53:33.000Z
Co-authored-by: Adam J. Stewart &lt;ajstewart426@gmail.com&gt;
diff --git a/segmentation_models_pytorch/encoders/__init__.py b/segmentation_models_pytorch/encoders/__init__.py
@@ -66,7 +66,8 @@ def get_encoder(name, in_channels=3, depth=5, weights=None, output_stride=32, **
     if name.startswith("timm-"):
         warnings.warn(
             "`timm-` encoders are deprecated and will be removed in the future. "
-            "Please use `tu-` equivalent encoders instead (see 'Timm encoders' section in the documentation)."
+            "Please use `tu-` equivalent encoders instead (see 'Timm encoders' section in the documentation).",
+            DeprecationWarning
         )
 
     # convert timm- models to tu- models

Original file line number	Diff line number	Diff line change
`@@ -66,7 +66,8 @@ def get_encoder(name, in_channels=3, depth=5, weights=None, output_stride=32, **`
`66`	`66`	`if name.startswith("timm-"):`
`67`	`67`	`warnings.warn(`
`68`	`68`	"`timm-` encoders are deprecated and will be removed in the future. "
`69`		- "Please use `tu-` equivalent encoders instead (see 'Timm encoders' section in the documentation)."
	`69`	+ "Please use `tu-` equivalent encoders instead (see 'Timm encoders' section in the documentation).",
	`70`	`+ DeprecationWarning`
`70`	`71`	`)`
`71`	`72`
`72`	`73`	`# convert timm- models to tu- models`