diff --git a/segmentation_models_pytorch/__init__.py b/segmentation_models_pytorch/__init__.py index c8209226..5cde6004 100644 --- a/segmentation_models_pytorch/__init__.py +++ b/segmentation_models_pytorch/__init__.py @@ -26,7 +26,9 @@ # Suppress the specific SyntaxWarning for `pretrainedmodels` warnings.filterwarnings("ignore", message="is with a literal", category=SyntaxWarning) -warnings.filterwarnings("ignore", message=r'"is" with \'str\' literal.*', category=SyntaxWarning) # for python >= 3.12 +warnings.filterwarnings( + "ignore", message=r'"is" with \'str\' literal.*', category=SyntaxWarning +) # for python >= 3.12 def create_model(