diff --git a/icevision/models/__init__.py b/icevision/models/__init__.py index 77e10f36a..0a6fa66f4 100644 --- a/icevision/models/__init__.py +++ b/icevision/models/__init__.py @@ -29,7 +29,7 @@ shell = session.__class__.__module__ # HACK: yolov5 breaks automatic setting of backend setting by notebook if shell in ["google.colab._shell", "ipykernel.zmqshell"]: - session.run_line_magic('matplotlib', 'inline') + session.run_line_magic("matplotlib", "inline") if SoftDependencies.mmseg: from icevision.models import mmseg