Import checksumdir with no alias

Signed-off-by: Martin Lyrå <[email protected]>
flyteorg · Oct 21, 2021 · c03c6fb · c03c6fb
1 parent 722c334
commit c03c6fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flytekit/tools/fast_registration.py b/flytekit/tools/fast_registration.py
@@ -4,7 +4,7 @@
 import tempfile as _tempfile
 from pathlib import Path as _Path
 
-import checksumdir as _dirhash
+import checksumdir
 
 from flytekit.interfaces.data import data_proxy as _data_proxy
 from flytekit.interfaces.data.data_proxy import Data as _Data
@@ -18,7 +18,7 @@ def compute_digest(source_dir: _os.PathLike) -> str:
     :param _os.PathLike source_dir:
     :return Text:
     """
-    return f"fast{_dirhash(source_dir, 'md5')}"
+    return f"fast{checksumdir.dirhash(source_dir, 'md5')}"
 
 
 def _write_marker(marker: _os.PathLike):