Merge pull request #794 from analysiscenter/fix/model-plot

Fix plot for model

Author: AlexeyKozhevin

batchflow/models/torch/base_mixins.py

@@ -6,10 +6,7 @@
 import torch
 
 from ...decorators import deprecated
-
-from ...utils_import import try_import
-plot = try_import(module='...plotter', package=__name__, attribute='plot',
-                  help='Try `pip install batchflow[image]`!')
+from ... import plot
 
 # Also imports `tensorboard`, if necessary