Merge pull request #65 from keisen/hotfixes/fixes-incorrect-error-mes…

…sage Hotfix: Improve some error messages
keisen · Jul 2, 2021 · ab21593 · ab21593
2 parents 8bc4c8f + 68b7367
commit ab21593
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name="tf-keras-vis",
-    version="0.7.0",
+    version="0.7.1",
     author="keisen",
     author_email="[email protected]",
     description="Neural network visualization toolkit for tf.keras",

diff --git a/tf_keras_vis/activation_maximization/__init__.py b/tf_keras_vis/activation_maximization/__init__.py
@@ -361,7 +361,8 @@ def _get_regularizers(self, regularizer):
         if legacy_regularizers is not None:
             warnings.warn(
                 "`tf_keras_vis.utils.regularizers.Regularizer` is deprecated. "
-                "Use tf_keras_vis.utils.regularizers.Regularizer instead.", DeprecationWarning)
+                "Use tf_keras_vis.activation_maximization.regularizers.Regularizer instead.",
+                DeprecationWarning)
             return legacy_regularizers
         else:
             regularizers = self._get_callables_to_apply_to_each_input(regularizer, "regularizers")
@@ -393,7 +394,7 @@ def _get_legacy_regularizers(self, regularizer):
                 if any(has_legacy):
                     raise ValueError(
                         "Legacy Regularizer instances (that inherits "
-                        "`tf_keras_vis.utils.regularizers.Regularizer`) must be "
+                        "`tf_keras_vis.utils.regularizers.LegacyRegularizer`) must be "
                         "passed to ActivationMaximization#__call__() "
                         "in the form of a instance or a list of instances. "
                         "Please modify the `regularizer` argument or "
@@ -409,7 +410,7 @@ def _get_legacy_regularizers(self, regularizer):
                         "the regularizer instance (that inherits "
                         "`tf_keras_vis.activation_maximization.regularizers.Regularizer`) "
                         "and legacy regularizer (that inherits "
-                        "`tf_keras_vis.utils.regularizers.Regularizer` can NOT be mixed."
+                        "`tf_keras_vis.utils.regularizers.LegacyRegularizer` can NOT be mixed."
                         f" regularizer: {regularizer}")
         elif isinstance(_regularizer, LegacyRegularizer):
             return listify(_regularizer)