fix save_logs.py (#2268)

Author: savingoyal

metaflow/mflog/save_logs.py

@@ -8,12 +8,12 @@
 from metaflow.util import Path
 from . import TASK_LOG_SOURCE
 
-from metaflow.tracing import cli_entrypoint
+from metaflow.tracing import cli
 
 SMALL_FILE_LIMIT = 1024 * 1024
 
 
-@cli_entrypoint("save_logs")
+@cli("save_logs")
 def save_logs():
     def _read_file(path):
         with open(path, "rb") as f: