remove import

Author: mufaddal-rohawala

src/sagemaker/workflow/_repack_model.py

@@ -33,8 +33,6 @@
 # repacking is some short-lived hackery, right??
 from distutils.dir_util import copy_tree
 
-from sagemaker.utils import check_tarfile_data_filter_attribute
-
 
 def repack(inference_script, model_archive, dependencies=None, source_dir=None):  # pragma: no cover
     """Repack custom dependencies and code into an existing model TAR archive
@@ -62,8 +60,7 @@ def repack(inference_script, model_archive, dependencies=None, source_dir=None):
         # extract the contents of the previous training job's model archive to the "src"
         # directory of this training job
         with tarfile.open(name=local_path, mode="r:gz") as tf:
-            check_tarfile_data_filter_attribute()
-            tf.extractall(path=src_dir, filter="data")
+            tf.extractall(path=src_dir)
 
         if source_dir:
             # copy /opt/ml/code to code/

src/sagemaker/workflow/_utils.py

@@ -36,6 +36,7 @@
     _save_model,
     download_file_from_url,
     format_tags,
+    check_tarfile_data_filter_attribute,
 )
 from sagemaker.workflow.retry import RetryPolicy
 from sagemaker.workflow.utilities import trim_request_dict
@@ -261,7 +262,8 @@ def _inject_repack_script_and_launcher(self):
                 download_file_from_url(self._source_dir, old_targz_path, self.sagemaker_session)
 
                 with tarfile.open(name=old_targz_path, mode="r:gz") as t:
-                    t.extractall(path=targz_contents_dir)
+                    check_tarfile_data_filter_attribute()
+                    t.extractall(path=targz_contents_dir, filter="data")
 
                 shutil.copy2(fname, os.path.join(targz_contents_dir, REPACK_SCRIPT))
                 with open(