add deprecation warnings for keras and tfv1 support (#1607)

src/sparseml/keras/__init__.py

@@ -18,6 +18,13 @@
 
 # flake8: noqa
 
+from sparseml.utils import deprecation_warning as _deprecation_warning
+
+
+_deprecation_warning(
+    "sparseml.keras is deprecated and will be removed in a future version",
+)
+
 from sparseml.analytics import sparseml_analytics as _analytics
 
 from .base import *

src/sparseml/tensorflow_v1/__init__.py

@@ -18,6 +18,13 @@
 
 # flake8: noqa
 
+from sparseml.utils import deprecation_warning as _deprecation_warning
+
+
+_deprecation_warning(
+    "sparseml.tensorflow_v1 is deprecated and will be removed in a future version",
+)
+
 from sparseml.analytics import sparseml_analytics as _analytics
 
 from .base import *

src/sparseml/utils/helpers.py

@@ -23,6 +23,7 @@
 import logging
 import os
 import sys
+import warnings
 from collections import OrderedDict
 from pathlib import Path
 from typing import Any, Callable, Dict, Iterable, List, Tuple, Union
@@ -64,6 +65,7 @@
     "tensors_export",
     "parse_optimization_str",
     "json_to_jsonl",
+    "deprecation_warning",
 ]
 
 
@@ -820,3 +822,12 @@ def json_to_jsonl(json_file_path: str, overwrite: bool = True):
         for json_line in json_data:
             json.dump(json_line, jsonl_file)  # append json line
             jsonl_file.write("\n")  # newline
+
+
+def deprecation_warning(message: str):
+    warnings.simplefilter("always", DeprecationWarning)
+    warnings.warn(
+        message,
+        category=DeprecationWarning,
+        stacklevel=2,
+    )